Software Engineer, Machine Learning Infrastructure

Job Overview

As a Machine Learning Software Engineer within the Dojo team at Tesla, you will play a pivotal role in enhancing the capabilities of our cutting-edge Dojo training accelerator. This position involves close collaboration with top-tier ML Researchers, Compiler, and Hardware Engineers to address unique challenges at the intersection of AI and ML training accelerators. Your expertise will be crucial in optimizing and scaling our neural network training infrastructure.

Key Responsibilities

Collaborate with machine learning researchers and engineers to run Full Self-Driving (FSD) models on our in-house ML training accelerator.
Profile performance of training workloads in our cluster, identify bottlenecks in and between CPU/Dojo code execution, and work on optimizing throughput and scalability within and across nodes to reduce convergence time.
Coordinate with the team managing the hardware cluster to maintain high availability and job throughput for machine learning tasks.
Integrate the training software into our continuous integration cluster to support metrics persistence across experiments, weekly/nightly neural network builds, and other unit/throughput tests.

Required Skills and Experience

Degree in Engineering, Computer Science, or equivalent experience with evidence of exceptional ability.
Practical experience programming in Python and/or C++.
Experience working with training frameworks, ideally PyTorch.
Proficient in system-level software, particularly hardware-software interactions and resource utilization.
Understanding of modern machine learning concepts and state-of-the-art deep learning.
Experience in profiling and optimizing CPU-accelerator interactions (pipelining compute/transfers, etc.).
DevOps experience, particularly in managing clusters of training nodes and filesystems for large amounts of training data.

Benefits

As a full-time Tesla employee, you are eligible for a comprehensive benefits package starting from day one, including:

Aetna PPO and HSA plans with $0 payroll deduction options.
Family-building, fertility, adoption, and surrogacy benefits.
Dental and vision plans with options for $0 paycheck contribution.
Company-paid HSA contributions, life, AD&D, short-term, and long-term disability insurance.
401(k) with employer match, Employee Stock Purchase Plans, and other financial benefits.
Employee Assistance Program, sick and vacation time, 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, and employee discounts and perks.

Compensation

The expected annual salary for this position ranges from €104,000 to €360,000, depending on experience and level, plus cash and stock awards and benefits. The total compensation package may vary based on market location, job-related knowledge, skills, and experience.

Join us at Tesla and contribute to building a sustainable future by enhancing our machine learning infrastructure and accelerating the world's transition to sustainable energy.

Benefits
Extracted with AI

Disability insurance
Commuter benefits
401(k)
Aetna PPO and HSA plans
Family-building benefits
Dental and vision plans
Healthcare and Dependent Care FSAs
LGBTQ+ care concierge services
Employee Stock Purchase Plans
Life and disability insurance
Employee Assistance Program
Sick and Vacation time
Back-up childcare
Voluntary benefits
Weight Loss and Tobacco Cessation Programs
Tesla Babies program
Employee discounts and perks

Similar jobs

Last update: 23 minutes ago

Applied Intuition

Software Engineer - Autonomous Driving

Join Applied Intuition as a Software Engineer in Munich to tackle autonomous driving challenges with top ADAS/AV programs.

15 days ago

Senior Backend Engineer C++

Join DeepL as a Senior Backend Engineer C++ to design and maintain scalable backend services using C++ and AI technologies.

6 days ago

Staff Software Engineer

Join Aiven as a Staff Software Engineer to develop cloud operations platforms using open-source technologies. Hybrid work in Berlin.

6 days ago

Senior Software Engineer - AWS, Python, Ruby on Rails

Join HeyJobs as a Senior Software Engineer to design scalable systems using AWS, Python, and Ruby on Rails in a dynamic team.

9 days ago

Machine Learning Engineer - LLM Fine-tuning and Performance

Join NVIDIA as a Machine Learning Engineer specializing in LLM fine-tuning and performance optimization. Work with cutting-edge ML technologies.

a month ago

Software Engineer II - Developer Experience

Join Elastic as a Software Engineer II in Developer Experience, focusing on test frameworks for Kibana. Remote work, competitive benefits.

6 days ago

Expert Machine Learning Engineer

Join Dataroots as an Expert Machine Learning Engineer to design and deliver AI-powered solutions, focusing on machine learning models.

6 days ago

Software Engineer - Cloud Applications and Python

Join Topicus as a Software Engineer in Arnhem to develop cloud applications using Python, REST APIs, and ETL processes for healthcare data services.

5 days ago

Senior Software Engineer - C#/.NET

Join TrueLayer as a Senior Software Engineer in Milan, working with C#, .NET, AWS, and Kubernetes to build scalable systems.

6 days ago

Senior Software Engineer - LLM

Join Snowflake as a Senior Software Engineer to build scalable machine learning platforms with LLMs, leveraging Python and TensorFlow.

a month ago

Staff Software Engineer, Data Platform

Join Personio as a Staff Software Engineer in Berlin to build scalable data platforms using Kafka, Kubernetes, and AWS. Drive innovation and excellence.

6 days ago