Engineering Manager - Model Performance
Are you passionate about advancing the frontiers of artificial intelligence while leading a team of exceptional engineers? We are looking for a Tech Lead Manager focused on ML performance and inference. 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. If you thrive in a fast-paced startup environment and are excited about both leadership and technical challenges, we want to hear from you.
EXAMPLE INITIATIVES
You'll get to work on these types of projects as part of our Model Performance team:
Baseten Embeddings Inference: The fastest embeddings solution available
The Baseten Inference Stack
Driving model performance optimization
Requirements
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.
BONUS POINTS
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.
Benefits
BENEFITS
Competitive compensation package.
This is a unique opportunity to be part of a rapidly growing startup in one of the most exciting engineering fields of our era.
An inclusive and supportive work culture that fosters learning and growth.
Exposure to a variety of ML startups, offering unparalleled learning and networking opportunities.
The application process will continue on the employer's website.
Location
San Francisco
Similar Jobs
Site Reliability Engineer (SRE)
Baseten
San Francisco
Systems Engineer (Digital Tools & Engineering) - Level 3
Lockheed Martin Space
Fort Worth, Texas; Orlando, Florida
[Hiring] Research Scientist Sr- RF Systems/Data Fusion
Lockheed Martin Space
Boulder, Colorado; King of Prussia, Pennsylvania; Littleton, Colorado
[Hiring] Design Process Integration Engineer - Level 2
Lockheed Martin Space
Fort Worth, Texas