Software Engineer, ML Performance, Dojo
Description For Software Engineer, ML Performance, Dojo
Tesla's Dojo Machine Learning team is seeking a Senior Software Engineer to optimize neural network training on their custom silicon supercomputer systems. This role offers a unique opportunity to work alongside the engineers who developed Tesla's custom FSD Computer. You'll be responsible for scaling and improving the performance of PyTorch-derived neural networks on Tesla's massively parallel Dojo accelerators. The position involves close collaboration with both the Autopilot software team and Dojo hardware team, making it perfect for engineers passionate about the intersection of machine learning, distributed systems, and hardware optimization. Tesla offers comprehensive benefits including competitive salary, equity, and extensive healthcare coverage. The role is based in Palo Alto, putting you at the heart of Tesla's AI innovation. This is a chance to directly impact the future of autonomous driving technology while working with cutting-edge custom hardware and software systems.
Responsibilities For Software Engineer, ML Performance, Dojo
- Understand and model the end-to-end training performance of the Autopilot SW team's Pytorch-derived neural networks on the Dojo system
- Develop software that scales and improves training performance based on your analysis of the bottlenecks
- Collaborate with the Dojo HW team to understand current HW architecture and propose future improvements
Requirements For Software Engineer, ML Performance, Dojo
- Degree in Engineering, Computer Science, or equivalent in experience
- Experience scaling neural network training systems or other large distributed systems
- Familiarity with the internals of PyTorch and/or JAX
- Performance analysis experience
- Experience coding parallel programs
- Able to work from Palo Alto office
Benefits For Software Engineer, ML Performance, Dojo
- Aetna PPO and HSA plans with $0 payroll deduction
- Family-building, fertility, adoption and surrogacy benefits
- Dental and vision plans with $0 paycheck contribution
- Company Paid HSA Contribution
- Healthcare and Dependent Care Flexible Spending Accounts (FSA)
- 401(k) with employer match
- Employee Stock Purchase Plans
- Company paid Basic Life, AD&D, short-term and long-term disability insurance
- Employee Assistance Program
- Sick and Vacation time
- Back-up childcare and parenting support resources
- Commuter benefits
- Employee discounts and perks program
Jobs Related To Tesla Software Engineer, ML Performance, Dojo
