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 enabling Tesla's neural networks to train efficiently on their in-house Dojo accelerators, focusing on performance optimization and scaling of PyTorch-derived neural networks.
The position combines cutting-edge machine learning infrastructure development with high-performance computing, offering exposure to Tesla's revolutionary autonomous driving technology. You'll work with both software and hardware teams, analyzing and improving training performance while contributing to future hardware architecture decisions.
This role offers competitive compensation ranging from $120,000 to $360,000 annually, plus additional cash and stock awards. Tesla provides comprehensive benefits including full medical coverage, 401(k) matching, and unique perks like the Tesla Babies program and family-building benefits.
The ideal candidate will bring strong experience in distributed systems, neural network training optimization, and parallel programming. You'll be part of a small, experienced team working on transformative technology that directly impacts Tesla's autonomous driving capabilities. This position requires working from the Palo Alto office, putting you at the heart of Tesla's AI development efforts.
Responsibilities For Software Engineer, ML Performance, Dojo
- Understand and model end-to-end training performance of Autopilot SW team's Pytorch-derived neural networks on the Dojo system
- Develop software that scales and improves training performance based on bottleneck analysis
- 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
- 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
- Weight Loss and Tobacco Cessation Programs
- Tesla Babies program
- Commuter benefits
- Employee discounts and perks program
Jobs Related To Tesla Software Engineer, ML Performance, Dojo
