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

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 4 days 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?

Jobs Related To Tesla Staff Software Engineer, ML Inference Compiler & Deployment, Optimus

Staff Machine Learning Engineer, Autobidder

Lead machine learning engineer position at Tesla, developing forecasting algorithms for energy storage systems and renewable assets, requiring 6+ years of Python experience.

Staff Performance Optimization Engineer, AI Infrastructure

Staff Performance Optimization Engineer position at Tesla, focusing on optimizing AI infrastructure and ML model training systems for autonomous driving and robotics applications.

Staff Software Engineer, Generative AI Platforms

Lead development of Tesla's Generative AI Platform, building cutting-edge applications and scalable solutions while working with advanced ML technologies.

Staff IC Packaging Mechanical Engineer, Dojo

Staff IC Packaging Mechanical Engineer position at Tesla working on Dojo supercomputer and self-driving hardware development.

Staff Performance Modeling Engineer, Dojo

Staff Performance Modeling Engineer position at Tesla, focusing on AI training infrastructure development and optimization with competitive compensation and benefits.