Tech Lead Manager, Baseten
Model Performance and Inference
$175-275k
+ Equity
ML application builder for data scientists
Open for applications
ML application builder for data scientists
21-100 employees
Open for applications
$175-275k
+ Equity
21-100 employees
Company mission
To make machine learning accessible to every organisation.
Role
Who you are
- This role is ideal for someone with a strong engineering background who is eager to lead and mentor a team while remaining hands-on with technology
- Bachelor’s, Master’s, or Ph.D. in Computer Science, Engineering, or a related field
- 5+ years of professional experience in software engineering, with at least 2 years in a technical leadership role
- Proven experience managing and mentoring teams of engineers
- Expertise in one or more programming languages, such as Python, C++, or Go
- In-depth understanding of ML model performance optimization, especially using libraries such as PyTorch, TensorRT, and CUDA
- Strong knowledge of containerization (Docker) and orchestration systems (Kubernetes)
- Experience with production-level AI/ML solutions, including scaling and deploying large models
- Ability to balance hands-on technical work with team leadership and project management
Desirable
- Experience enhancing the performance of large language models (LLMs) or similar AI systems
- Familiarity with LLM optimization techniques such as quantization, speculative decoding, or continuous batching
- Deep knowledge of GPU architecture and performance tuning
- Previous experience in a high-growth startup environment
What the job involves
- We are looking for a Tech Lead Manager focused on ML model performance and inference
- Lead, mentor, and manage a team of engineers focused on developing and optimizing ML model inference and performance
- Oversee technical strategy and architecture decisions, driving improvements across our engineering organization
- Collaborate with cross-functional teams to ensure seamless integration and scalability of ML models in production environments
- Dive into the codebase of frameworks like TensorRT, PyTorch, CUDA, and others to identify and solve complex performance bottlenecks
- Drive the development and deployment of large-scale optimization techniques for various ML models, especially large language models (LLMs)
- Own the full lifecycle of projects from inception through delivery, including planning, execution, and resource management
- Foster a collaborative, inclusive team environment that encourages continuous learning and growth
Our take
As with its mathematical namesake, Baseten focuses on simplifying the complex. The San Francisco startup focuses on simplifying the integration of artificial intelligence into businesses. The tools that Baseten built demolish the need for businesses to hire specialised engineers, utilising the huge strides in artificial intelligence that we’ve seen in the last few years without adding workload to businesses.
With serious funding, Baseten plans to use its investment to expand its engineering and go-to-market teams. This is part of its focus on helping companies implement the tools that they have at their disposal, a long-overlooked aspect of machine learning that many businesses have been eager to champion, but uncertain on how to actually use.
Simplifying the complex world of artificial intelligence for business is no mean feat but the three founders, who met whilst working as engineers at Gumroad, have a lot of experience in how to navigate this gap in the market.
Kirsty
Company Specialist at Welcome to the Jungle
Insights
Few candidates hear
back within 2 weeks
Company
Funding (last 2 of 3 rounds)
Mar 2024
$40m
SERIES B
Apr 2022
$12m
SERIES A
Total funding: $60m
Company benefits
- Remote-first work environment. The Baseten team is welcome to work from wherever they want; fully remote, in our San Francisco office, or a mix of both. Today, our team (including our founding team) is spread across the United States, Canada, and Armenia. We provide a $1,000 stipend for you to make your home-office comfortable and productive
- Regular in-person team summits. We get together as a team three times a year to plan, workshop, and most importantly, get to know each other better
- Unlimited PTO. We ask that everyone take at least 4 weeks of vacation. And we have a company-wide break between Christmas and New Year's Day
- Full healthcare coverage. Medical, dental and vision insurance for you and your family
- Paid parental leave. 16-weeks fully paid parental leave (adoptive and non-birth parents included) and flexibility with schedules while returning to work
- Company-sponsored 401(k) for you to contribute to
- Learning and development budget. We encourage you to take classes, attend conferences, and invest in your craft and we’ll cover expenses to make it happen
Company HQ
Union Square, San Francisco, CA
Articles
Leadership
Tuhin Srivastava
(Co-Founder)Previously Co-Founder at Shape, Machine Learning at Gumroad and Skulpt.
Amir Haghighat
(Co-Founder)Previously Head of Engineering at Gumroad, Software Engineer at Yelp and Engineering Manager at Clover Health.
Phillip Howes
(Co-Founder)Previously Co-Founder at Shape, Machine Learning Engineer at Gumroad and Research Assistant at the University of Sydney.
Salary benchmarks
We don't have enough data yet to provide salary benchmarks for this role.
Submit your salary to help other candidates with crowdsourced salary estimates.
Share this job
View 11 more jobs at Baseten