Sr. Machine Learning Engineer, Autopilot Hardware

Tesla - Austin, TX

Full-time - Mid Level

Austin, TX

Transportation Equipment Manufacturing

About the position

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. This role requires a deep understanding of neural network architecture, compilers, and the mapping of various operations onto the hardware. The ideal candidate is an experienced machine learning engineer with a strong understanding of GPUs and AI accelerators, and is comfortable working rapidly in a small-team environment. You will be expected to analyze and deploy state-of-the-art neural networks on GPU and AI accelerators, while also developing ML compiler support for new hardware features. Building high-level performance models and analyzing performance will be a key part of your responsibilities, as well as proposing new hardware features backed by performance estimates. Debugging performance bottlenecks and issues in hardware or simulation will also be a critical aspect of your role.

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

Benefits

• Aetna PPO and HSA plans with 2 medical plan options and $0 payroll deduction
• Family-building, fertility, adoption and surrogacy benefits
• Dental (including orthodontic coverage) and vision plans with $0 paycheck contribution options
• Company Paid 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
• 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 including 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