Sr. Machine Learning Engineer, Autopilot Hardware Full-time

As a member of the Autopilot hardware team, you will be responsible for enabling Tesla’s neural networks on our upcoming in-house custom Machine Learning Silicon. Machine learning engineers are expected to understand neural network architecture, compilers and mapping of various operations on to the hardware. The ideal candidate is an experienced ML engineer, with strong understanding of GPUs and AI accelerators, and is comfortable working rapidly in a small-team environment.

Responsibilities

• Analyzing and deploying state-of-the-art neural networks on GPU, AI accelerator and performance models
• Develop ML compiler support for new hardware features
• Build high-level performance models and analyze performance
• Propose new hardware features, backed by performance estimates
• Debug performance bottlenecks and issues in hardware or simulation

Requirements

• Bachelor’s Degree in Computer Science, Computer Engineering or Electrical Engineering or evidence of equivalent experience and exceptional skills
• 5+ years of experience in machine learning frameworks, and optimizing neural network performance on hardware
• Comfortable with C++ and python
• Good understanding of GPU architecture and AI accelerators
• Ability to root cause performance issues and debug esoteric symptoms
• Go-getter attitude and capable of pursuing tough engineering challenges to final solutions

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