Sr. Machine Learning Engineer, Self-Driving Hardware

Tesla - Austin, TX

Full-time - Mid Level

Austin, TX

Transportation Equipment Manufacturing

About the position

As a member of the Self-Driving hardware team at Tesla, you will play a crucial role in enabling the company's neural networks on our upcoming in-house custom Machine Learning Silicon. This position requires a deep understanding of neural network architecture, compilers, and the mapping of various operations onto hardware. The ideal candidate will be an experienced Machine Learning engineer who possesses 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. Your role will involve building high-level performance models, analyzing performance, and proposing new hardware features backed by performance estimates. Additionally, you will be responsible for debugging performance bottlenecks and issues in hardware or simulation, ensuring that the systems operate at optimal efficiency.

Responsibilities

• Analyzing and deploying state-of-the-art neural networks on GPU, AI accelerator and performance models
• Developing ML compiler support for new hardware features
• Building high-level performance models and analyzing performance
• Proposing new hardware features, backed by performance estimates
• Debugging 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