Staff Software Engineer, ML Inference Compiler & Deployment, Optimus

Tesla develops advanced electric vehicles and AI systems, including autonomous driving technology and humanoid robots.
$120,000 - $360,000
Machine Learning
Staff Software Engineer
In-Person
5,000+ Employees
8+ years of experience
AI · Automotive · Robotics
This job posting may no longer be active. You may be interested in these related jobs instead:
Staff Machine Learning Engineer, Autobidder

Staff Machine Learning Engineer position at Tesla working on Autobidder platform for energy storage optimization and trading.

Staff, Data Scientist - Machine Learning Engineer | Catalog Trust & Safety Team

Staff Data Scientist role at Walmart focusing on Machine Learning and AI for marketplace trust and safety, offering competitive pay and comprehensive benefits in Bentonville, AR.

Lead Technical Program Manager- AI/ML

Lead Technical Program Manager position at JPMorgan Chase focusing on AI/ML initiatives, offering competitive compensation and comprehensive benefits.

Research Scientist 4 - Globalization

Senior Research Scientist position at Netflix focusing on ML and generative speech technologies for global content localization.

Staff Machine Learning Engineer - Maps

Staff Machine Learning Engineer position at Uber's Amsterdam office, focusing on developing advanced mapping solutions using ML technologies.

Description For Staff Software Engineer, ML Inference Compiler & Deployment, Optimus

Tesla is seeking a Staff Software Engineer to join their Autonomy teams, focusing on ML Inference Compiler & Deployment for Optimus, their humanoid robot project. This role is at the intersection of AI, robotics, and autonomous systems, working on one of the world's most advanced AI platforms.

The position involves deep work with Tesla's AI inference stack, which powers both their Autopilot system and Optimus robot. You'll be collaborating with AI Engineers and Hardware Engineers to optimize neural network performance and efficiency. The role is highly impactful, as your work will directly affect millions of Tesla vehicles and the Optimus platform.

This is an exceptional opportunity for someone passionate about cutting-edge AI technology and its real-world applications. You'll work with a purpose-driven inference stack development process, where your optimizations and improvements have immediate, measurable impacts on production systems. The position offers access to Tesla's proprietary MLIR compiler and runtime architecture, allowing for innovative approaches to model optimization.

The role combines deep technical expertise in C++ and Python with specialized knowledge in machine learning compilers and frameworks. You'll be part of a team that's pushing the boundaries of AI hardware and software co-design, working on massively-parallel systems, and contributing to the evolution of autonomous systems and robotics.

Tesla offers comprehensive benefits including competitive salary, equity opportunities, and extensive health coverage. The company's culture promotes innovation, direct impact, and the opportunity to work on technologies that are transforming multiple industries simultaneously.

Last updated 2 months ago

Responsibilities For Staff Software Engineer, ML Inference Compiler & Deployment, Optimus

  • Develop, optimize, maintain, and debug the AI inference stack for Optimus
  • Co-Design models for efficient execution on AI HW
  • Collaborate with AI/runtime/OS/driver teams for efficient compilation and execution
  • Analyze and debug functional and performance issues on massively-parallel systems
  • Maintain compatibility with latest ML/compiler community developments

Requirements For Staff Software Engineer, ML Inference Compiler & Deployment, Optimus

Python
  • Proficient C/C++ programming (C++14/17/20)
  • Basic Python proficiency
  • Prior experience working with ML compilers (e.g. MLIR, XLA, TensorRT)
  • Experience with ML frameworks and ML architecture (e.g. PyTorch, JAX, TF)
  • Degree in Engineering, Computer Science, or equivalent experience

Benefits For Staff Software Engineer, ML Inference Compiler & Deployment, Optimus

Medical Insurance
Dental Insurance
Vision Insurance
401k
Mental Health Assistance
Parental Leave
Commuter Benefits
  • Medical plans with $0 payroll deduction
  • Family-building, fertility, adoption and surrogacy benefits
  • Dental and vision plans
  • Company Paid HSA Contribution
  • Healthcare and Dependent Care FSA
  • 401(k) with employer match
  • Employee Stock Purchase Plans
  • Company paid Basic Life, AD&D, disability insurance
  • Employee Assistance Program
  • Sick and Vacation time
  • Back-up childcare and parenting support
  • Commuter benefits
  • Employee discounts and perks program

Interested in this job?