Software Engineer, ML Compiler, Dojo

Tesla is an automotive and clean energy company that develops electric vehicles and AI systems.
$104,000 - $360,000
Machine Learning
Senior Software Engineer
In-Person
5,000+ Employees
5+ years of experience
AI · Automotive

Description For Software Engineer, ML Compiler, Dojo

Tesla is seeking a Software Engineer for their ML Compiler team working on the Dojo project, their custom-silicon supercomputer system. This role involves working with a small team of experienced developers to automate the compilation of PyTorch-derived neural network graphs for Tesla's custom FSD computer. The position requires expertise in compiler development and C++ programming, offering an opportunity to work on cutting-edge AI hardware and software integration. The role involves collaboration with both hardware and Autopilot software teams, making it ideal for someone who enjoys working at the intersection of machine learning, compiler optimization, and hardware architecture. Tesla offers comprehensive benefits including competitive salary, equity, and extensive healthcare coverage. This position is central to Tesla's AI development efforts and offers the chance to work on technology that will directly impact the future of autonomous driving.

Last updated 2 months ago

Responsibilities For Software Engineer, ML Compiler, Dojo

  • Take ownership of compiler areas based on skills/interests/needs
  • Develop algorithms to improve performance and reduce compiler overhead
  • Debug functional and performance issues on massively parallel systems
  • Collaborate with Dojo HW team on hardware architecture
  • Work with Autopilot SW team for training transition from GPU to Dojo

Requirements For Software Engineer, ML Compiler, Dojo

  • Degree in Engineering, Computer Science, or equivalent experience
  • Prior industry or research experience within compiler development
  • Comfortable with C++ and assembly code

Benefits For Software Engineer, ML Compiler, Dojo

401k
Medical Insurance
Dental Insurance
Vision Insurance
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
  • Life, AD&D, short-term and long-term disability insurance
  • Employee Assistance Program
  • Sick and Vacation time
  • Back-up childcare
  • Commuter benefits
  • Employee discounts

Interested in this job?

Jobs Related To Tesla Software Engineer, ML Compiler, Dojo

Software Engineer, Generalist, AI Inference

Senior Software Engineer position at Tesla focusing on AI inference optimization for autonomous driving and robotics applications.

Fullstack Software Engineer, Machine Learning Platform, AI Infrastructure

Senior Software Engineer role at Tesla working on ML infrastructure for autonomous vehicles and robotics, offering $116K-$360K salary plus benefits.

Sr. Deep Learning Vision Engineer, Tesla Bot

Senior Deep Learning Vision Engineer position at Tesla, focusing on developing advanced AI algorithms for the Tesla Bot humanoid robot project.

Sr. Machine Learning Engineer, Autobidder

Sr. Machine Learning Engineer for Tesla's Autobidder team, developing forecasting algorithms for energy storage and renewable assets.

Software Engineer III - ML, Python

Senior Software Engineer role at JPMorgan Chase focusing on ML fraud prevention platform development, offering competitive salary and comprehensive benefits.