Machine Learning Compiler Engineer, Self-Driving Hardware
Description For Machine Learning Compiler Engineer, Self-Driving Hardware
Tesla is seeking a Machine Learning Compiler Engineer to join their Self-Driving Hardware team. This role focuses on enabling Tesla's neural networks on their upcoming in-house custom Machine Learning Silicon. You'll be part of a small, experienced team working on automating the compilation of PyTorch-derived neural network graphs for Tesla's custom FSD computer. The position requires expertise in compiler development, with responsibilities including hardware silicon support, architecture improvement, and performance optimization. The ideal candidate should have strong programming skills in Python and C++, understanding of GPU architecture, and compiler development experience. Tesla offers comprehensive benefits including healthcare, 401(k) with matching, stock purchase options, and various family support benefits. This is an opportunity to contribute to cutting-edge autonomous vehicle technology at one of the world's leading electric vehicle manufacturers.
Responsibilities For Machine Learning Compiler Engineer, Self-Driving Hardware
- Take ownership of compiler areas based on skills/interests/needs
- Bring up new hardware silicon and add support in the compiler for hardware features
- Collaborate with design team to understand architecture and propose improvements
- Develop algorithms to improve performance while decreasing power
Requirements For Machine Learning Compiler Engineer, Self-Driving Hardware
- Degree in Engineering, Computer Science, or equivalent experience
- Prior industry or research experience in compiler development
- Comfortable with Python and C++
- Good understanding of GPU architecture and AI accelerators
- Excellent problem solving and debugging skills
Benefits For Machine Learning Compiler Engineer, Self-Driving Hardware
- 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 FSA
- LGBTQ+ care concierge services
- 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
- Commuter benefits
- Employee discounts and perks program
