Machine Learning Compiler Engineer, Self-Driving Hardware Full-time

As a member of the Self-Driving hardware team, you will be responsible for enabling Tesla’s neural networks on our upcoming in-house custom Machine Learning Silicon. Join a small team of experienced developers in automating the compilation of PyTorch-derived neural network graphs into programs that run on next-generation Tesla’s custom FSD computer. The ideal candidate is an experienced compiler engineer comfortable working rapidly in a small-team environment.

Responsibilities

• Take ownership of a few areas of the compiler (flexible, based on skills/interests/needs)
• Bring up new hardware silicon and add support in the compiler for these hardware features
• Collaborate with the design team to understand current architecture and propose future improvements
• Develop algorithms to improve performance, while decreasing power

Requirements

• Degree in Engineering, Computer Science, or equivalent in experience and evidence of exceptional ability
• Prior industry or research experience within compiler development
• Comfortable with Python and C++
• Experience with MLIR and LLVM is a plus
• Good understanding of GPU architecture and AI accelerators
• Excellent problem solving and debugging skills

Compensation and Benefits
Benefits

Along with competitive pay, as a full-time Tesla employee, you are eligible for the following benefits at day 1 of hire:

• Aetna PPO and HSA plans > 2 medical plan options with $0 payroll deduction
• Family-building, fertility, adoption and surrogacy benefits
• Dental (including orthodontic coverage) and vision plans, both have options with a $0 paycheck contribution
• Company Paid (Health Savings Account) HSA Contribution when enrolled in the High Deductible Aetna medical plan with HSA
• Healthcare and Dependent Care Flexible Spending Accounts (FSA)
• LGBTQ+ care concierge services
• 401(k) with employer match, Employee Stock Purchase Plans, and other financial benefits
• Company paid Basic Life, AD&D, short-term and long-term disability insurance
• Employee Assistance Program
• Sick and Vacation time (Flex time for salary positions), and Paid Holidays
• Back-up childcare and parenting support resources
• Voluntary benefits to include: critical illness, hospital indemnity, accident insurance, theft & legal services, and pet insurance
• Weight Loss and Tobacco Cessation Programs
• Tesla Babies program
• Commuter benefits
• Employee discounts and perks program