Software Engineer, Generalist, AI Infrastructure

Tesla is an automotive and clean energy company that develops electric vehicles and AI-powered autonomous driving technology.
$104,000 - $360,000
Machine Learning
Mid-Level Software Engineer
In-Person
3+ years of experience
AI · Automotive

Description For Software Engineer, Generalist, AI Infrastructure

Tesla is seeking a Software Engineer to join their Autopilot team, focusing on AI infrastructure. This role is crucial for developing and optimizing the neural network training infrastructure that powers Tesla's self-driving capabilities. You'll be working at the intersection of machine learning and systems engineering, writing robust Python code, managing training pipelines, and optimizing performance across GPU clusters.

The position offers an exciting opportunity to work on cutting-edge AI technology in the automotive industry, directly contributing to Tesla's autonomous driving mission. You'll collaborate with machine learning scientists and infrastructure teams to scale and improve training systems for production models and experiments.

This role is perfect for someone who combines strong software engineering skills with deep understanding of machine learning systems and hardware optimization. You'll be responsible for critical infrastructure that enables Tesla's AI advancement, working with modern frameworks like PyTorch and handling large-scale training operations.

The role offers competitive compensation including both cash and stock awards, comprehensive benefits, and the chance to work on revolutionary technology that's transforming transportation. You'll be part of Tesla's mission to accelerate the world's transition to sustainable energy while working with some of the most advanced AI systems in the automotive industry.

Last updated 3 months ago

Responsibilities For Software Engineer, Generalist, AI Infrastructure

  • Write robust Python software code in machine learning training repository
  • Integrate training software into continuous integration cluster
  • Profile performance of training software and optimize throughput
  • Coordinate with hardware cluster team for ML job throughput
  • Support machine learning scientists in data fetching, preprocessing, and training runs

Requirements For Software Engineer, Generalist, AI Infrastructure

Python
Linux
  • Practical experience programming in Python and/or C/C++
  • Proficient in system-level software
  • Understanding of modern machine learning concepts
  • Experience working with training frameworks, ideally PyTorch
  • Experience scaling neural network training jobs across GPU clusters
  • Experience programming in CUDA
  • Profiling and optimizing CPU-GPU interactions
  • DevOps experience with training clusters

Benefits For Software Engineer, Generalist, AI Infrastructure

Medical Insurance
Dental Insurance
Vision Insurance
401k
Parental Leave
Commuter Benefits
  • Medical insurance with $0 payroll deduction options
  • Family-building, fertility, adoption and surrogacy benefits
  • Dental and vision plans
  • HSA with company contribution
  • Healthcare and Dependent Care FSA
  • 401(k) with employer match
  • Employee Stock Purchase Plans
  • Life, AD&D, short-term and long-term disability insurance
  • Employee Assistance Program
  • Sick and Vacation time
  • Back-up childcare
  • Commuter benefits
  • Employee discounts

Interested in this job?

Jobs Related To Tesla Software Engineer, Generalist, AI Infrastructure

Software Validation Engineer, Integration, AI Platforms

Software Validation Engineer role at Tesla, focusing on AI platform integration for autonomous vehicles and robotics systems.

Software Engineer, Machine Learning Infrastructure, Tesla Bot

Build and maintain machine learning infrastructure for Tesla Bot, focusing on neural network architecture, training optimization, and model deployment.

AI/ML Engineer II

AI/ML Engineer II position at JPMorgan Chase focusing on AWS Cloud, Databricks ML, and Gen AI platform development within Consumer & Community Banking division.

Software Engineer II, Customer eXperience Impressions (CXI)

Software Engineer position at Amazon's CXI team developing ML systems to detect and fix shopping experience issues, offering competitive pay and benefits.

Software Development Engineer II - DSO, Demand Science Optimization (DSO)

Software Development Engineer II position at Amazon's DSO team, focusing on ML-driven demand forecasting and supply management for Amazon Devices.