Baseten
The ML Application Builder for data scientists.
machine learning • developer tools • software engineering • artificial intelligence
11 - 50
Tech Lead Manager - Model Performance
6 days ago
🏢 In-office - San Francisco
Baseten
The ML Application Builder for data scientists.
machine learning • developer tools • software engineering • artificial intelligence
11 - 50
Description
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.
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.
Apply Now
