Baseten에서 ML 모델 성능 최적화 팀을 이끌 Tech Lead Manager를 채용합니다. PyTorch, CUDA, TensorRT 등 프레임워크에 대한 깊은 이해와 5년 이상의 소프트웨어 엔지니어링 경력, 2년 이상의 리더십 경험이 필수입니다. 대규모 AI 모델의 추론 성능을 개선하고 팀의 기술 전략을 주도하는 핵심 역할을 수행하게 됩니다.
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 $1.5B Series F, led by Altimeter Capital, Conviction Partners, and Spark Capital. Join us and help build the platform engineers turn to to ship AI products.
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.
You'll get to work on these types of projects as part of our Model Performance team:
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.
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.
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.
Competitive compensation, including meaningful equity.
100% coverage of medical, dental, and vision insurance for employee and dependents
Flexible PTO policy including company wide Winter Break (our offices are closed from Christmas Eve to New Year's Day!)
Paid parental leave
Fertility and family-building stipend through Carrot
Company-facilitated 401(k)
Exposure to a variety of ML startups, offering unparalleled learning and networking opportunities.
Apply now to embark on a rewarding journey in shaping the future of AI! If you are a motivated individual with a passion for machine learning and a desire to be part of a collaborative and forward-thinking team, we would love to hear from you.
At Baseten, we are committed to fostering a diverse and inclusive workplace. We provide equal employment opportunities to all employees and applicants without regard to race, color, religion, gender, sexual orientation, gender identity or expression, national origin, age, genetic information, disability, or veteran status.
We are an Equal Opportunity Employer and will consider qualified applicants with criminal histories in a manner consistent with applicable law (by example, the requirements of the San Francisco Fair Chance Ordinance, where applicable).
