Machine Learning Performance Software Engineer, Autopilot AI (Dojo)

Tesla
Apply Now

Job Description

As a member of the Dojo machine learning team, you will be responsible for enabling Tesla’s neural networks to train efficiently on our upcoming in-house custom-silicon supercomputer systems. Join a small team of experienced developers in optimizing and scaling the deployment of our Pytorch-derived neural networks on Tesla's custom massively-parallel Dojo accelerators. Work with many of the same great engineers who delivered Tesla’s custom FSD computer. The ideal candidate has experience with writing software for large distributed systems.

Responsibilities:

  • Understand and model the end to end training performance of the Autopilot SW team's Pytorch-derived neural networks on the Dojo system.
  • Develop software that scales and improves training performance based on your analysis of bottlenecks
  • Collaborate with the Dojo HW team to understand current HW architecture and propose future improvements

Requirements:

  • BS in relevant field (Computer Science, Computer Engineering, etc.) or relevant work experience 
  • Comfortable with C++ and Python
  • Capable of delivering results with minimal oversight
  • Good communication skills

Nice to Have:

  • Experience scaling neural network training systems or other large distributed systems
  • Familiarity with the internals of Pytorch
  • Performance analysis experience
  • Experience coding parallel programs
  • Able to work from Deer Creek (Palo Alto) office
  • Ready to start soon

Company Info.

Tesla

Tesla, Inc., headquartered in Austin, Texas, is an American multinational corporation specializing in electric vehicle design and manufacturing, as well as the production of stationary battery energy storage solutions spanning from residential to grid-scale applications. Additionally, Tesla offers a range of solar panels, solar shingles, and complementary products and services within the clean energy sector.

  • Industry
    Automotive,Energy,Autonomous technology
  • No. of Employees
    127,000
  • Location
    13101 Tesla Road, Austin, TX, USA
  • Website
  • Jobs Posted

Get Similar Jobs In Your Inbox

Tesla is currently hiring Machine Learning Performance Engineer Jobs in Palo Alto, CA, USA with average base salary of $120,000 - $190,000 / Year.

Similar Jobs View More