About the role
ABOUT BASETEN
Baseten powers mission-critical inference for the world's most dynamic AI companies, like Cursor, Notion, OpenEvidence, Abridge, Clay, Gamma and Writer. By uniting applied AI research, flexible infrastructure, and seamless developer tooling, we enable companies operating at the frontier of AI to bring cutting-edge models into production. We're growing quickly and recently raised our $300M Series E, backed by investors including BOND, IVP, Spark Capital, Greylock, and Conviction. Join us and help build the platform engineers turn to to ship AI products.
THE ROLE
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
RESPONSIBILITIES
- 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.