Machine Learning Compiler Engineer, Self-Driving Hardware

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

Employers have access to artificial intelligence language tools (“AI”) that help generate and enhance job descriptions and AI may have been used to create this description. The position description has been reviewed for accuracy and Dice believes it to correctly reflect the job opportunity.