Fullstack Software Engineer, Machine Learning Platform, AI Infrastructure

What to Expect

As a Software Engineer within the Autopilot AI Infrastructure team, you will work on reinforcing, optimizing, and scaling our infrastructure components supporting AI research activities for Autopilot and the Optimus.

At the core of our autonomy capabilities are neural networks that the research team is designing to train on very large amounts of data, across large-scale GPU clusters and our supercomputer Dojo. Robustly training these models at scale and in the shortest amount of time is critical to our mission.

We are building out the Machine Learning Platform that our engineers and leadership use to schedule, manage and monitor machine learning experiments, data pipelines and artifacts.

What You'll Do

• Architect and implement scalable, user-friendly tools for AI workflows to track and visualize the lifecycle of machine learning experiments and models
• Build robust tools and infrastructure to improve the machine learning team's velocity. This includes training and evaluation code in Python to back-end and front-end work in JavaScript
• Collaborate closely with ML engineers to ensure tools are aligned with research needs
• Design dashboards to provide real-time insights into performance and progress for our ML engineers and leadership
• Coordinate required hardware resources with the team managing the cluster hardware to maintain high availability

What You'll Bring

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 :

2 medical plan options with $0 payroll deduction

Expected Compensation $124,000 - $420,000 / annual salary + cash and stock awards + benefits

Pay offered may vary depending on multiple individualized factors, including market location, job-related knowledge, skills, and experience. The total compensation package for this position may also include other elements dependent on the position offered. Details of participation in these benefit plans will be provided if an employee receives an offer of employment.