Software Engineer, ML Performance, Dojo

Job Description

As a member of the Dojo Machine Learning team, you will be responsible for enabling Tesla's neural networks to train efficiently on custom silicon supercomputer systems. You will optimize and scale the deployment of PyTorch-derived neural networks on Tesla's Dojo accelerators, collaborating with engineers to improve performance.

Requirements

• Degree in Engineering, Computer Science, or equivalent experience
• Experience scaling neural network training systems or large distributed systems
• Familiarity with PyTorch and/or JAX
• Performance analysis experience
• Experience coding parallel programs
• Able to work from Palo Alto office

Responsibilities

• Understand and model training performance of neural networks
• Develop software to scale and improve training performance
• Collaborate with the Dojo HW team to propose improvements to current HW architecture.

Benefits

• Benefits include Aetna PPO and HSA plans with $0 payroll deduction, family-building benefits, dental and vision plans with $0 contribution, flexible spending accounts, 401(k) with employer match, company-paid life insurance, sick and vacation time, and employee discounts.