Engineering Manager - Model Performance Job at BaseTen Labs, Inc., San Francisco, CA

aDN6NmltK2VSSnZOaGZYbnQ2OVprT1hr
  • BaseTen Labs, Inc.
  • San Francisco, CA

Job Description

ABOUT BASETEN

Join our dynamic team at Baseten, where we’re revolutionizing AI deployment with cutting-edge inference infrastructure. Backed by premier investors such as IVP , Spark Capital , Greylock , and Conviction , we’re trusted by leading enterprises and AI-driven innovators—including Descript , Bland.ai , Patreon , Writer , and Robust Intelligence —to deliver top-tier performance, security, and reliability for their production workloads. With our recent $75 million Series C funding, we’re poised to accelerate our mission to make AI accessible across all products. If you’re passionate about tackling impactful challenges and building transformative solutions from the ground up, we invite you to join us on this exciting journey!

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.

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. 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.

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:

Competitive compensation package (Unlimited PTO, 401k, covered healthcare premiums). An opportunity to lead a talented engineering team at a rapidly growing startup in the machine learning space. Inclusive and supportive work culture with ample opportunities for professional development. Exposure to a wide range of ML use cases, offering unmatched learning and networking potential. #J-18808-Ljbffr BaseTen Labs, Inc.

Job Tags

Similar Jobs

LIV Golf

Director, Corporate Communications Job at LIV Golf

Location: New York, NYReporting To: Vice President, CommunicationsContract Term: Full time, permanentAbout LIV GolfThe LIV Golf League...  ...the world through golf.Role ObjectiveThe Director, Corporate Communications will be responsible for a variety of workflows... 

Pilot Flying J

Taco Bell Cashier Job at Pilot Flying J

 ...guests journey - a great one! Also, there are a number of opportunities to work in other roles within our travel centers and restaurants so while we may be hiring for a specific role, we always look to train and offer experience for other roles we have. We hope your... 

Maid Brigade of Prosper

Buscamos personas para nuestra familia de MAID BRIGADE. Job at Maid Brigade of Prosper

 ...donde se trata en una manera justa? donde esta parte de un equipo? donde puede ver los resultados de su trabajo dura? donde puede hacer una diferencia para otras...  ...! Podemos ensearle las habilidades valiosas de limpieza profesional y servicio al cliente que necesita para... 

DomestiCare Staffing

Housekeeper Job at DomestiCare Staffing

A family in Boca Raton area is looking for a housekeeper/cook. They need someone who is a great cook, can follow directions and is very...  ...Salary: $30-$35/hr/ Will consider medical insurance stipend Living Accommodations: Live Out Vacation: Yes Health Insurance... 

Wynne Transport Service, Inc.

OTR Tanker Truck Driver Job at Wynne Transport Service, Inc.

 ...Attention Florida Truck Drivers! OTR CDL A Tank Truck Solo Drivers needed to haul liquid bulk general hazmat products across 48 states. Positions Available: Company = Up to $0.68 a practical mile (loaded and empty + supplemental pay) or Median $1,650 Per Week...