Software Engineer, Generalist, AI Inference

Tesla is an automotive and technology company that develops electric vehicles and AI solutions for autonomous driving and robotics.
$104,000 - $360,000
Machine Learning
Senior Software Engineer
In-Person
5,000+ Employees
5+ years of experience
AI · Automotive · Robotics

Description For Software Engineer, Generalist, AI Inference

Tesla is seeking a Senior Software Engineer specializing in AI Inference to join their cutting-edge team. This role sits at the intersection of machine learning and systems engineering, focusing on productionalizing ML models for Tesla's Autopilot and Optimus projects. The position involves working with large neural networks and optimizing them for edge devices using various compute platforms (CPU/GPU/AI ASIC).

The role is highly technical and multidisciplinary, requiring expertise in both machine learning frameworks and low-level systems programming. You'll be responsible for building robust AI infrastructure that enables seamless training and deployment of neural networks, working with state-of-the-art models, and optimizing their performance for resource-constrained environments.

As part of Tesla's AI team, you'll collaborate with AI scientists and compiler engineers to tackle complex challenges in model compression and efficient inference. The position offers competitive compensation, comprehensive benefits, and the opportunity to work on revolutionary technology that's shaping the future of autonomous vehicles and robotics.

The ideal candidate should have strong programming skills in Python and modern C++, deep understanding of ML frameworks like PyTorch, experience with CUDA programming, and a solid foundation in computer systems architecture. This is an excellent opportunity for someone passionate about pushing the boundaries of AI deployment in real-world applications.

Last updated 2 months ago

Responsibilities For Software Engineer, Generalist, AI Inference

  • Build robust AI frameworks to lower neural networks to edge devices
  • Build robust AI infrastructure to train and fine-tune networks for Autopilot and Optimus on large GPU clusters
  • Deploy state-of-the-art neural networks on heterogenous compute
  • Collaborate with AI scientists and compiler engineers to compress large models
  • Design and implement custom GPU kernels for efficient training and post-processing

Requirements For Software Engineer, Generalist, AI Inference

Python
Kubernetes
  • Proficiency with Python and C++, including modern C++ (14/17/20)
  • Proficiency with PyTorch or another machine learning framework
  • Proficiency with training and deploying neural networks for real-world AI
  • Proficiency with computer systems and computer architecture
  • Experience with CUDA

Benefits For Software Engineer, Generalist, AI Inference

Medical Insurance
Dental Insurance
Vision Insurance
401k
Mental Health Assistance
Parental Leave
Commuter Benefits
  • Aetna PPO and HSA plans with $0 payroll deduction
  • Family-building, fertility, adoption and surrogacy benefits
  • Dental and vision plans with $0 paycheck contribution
  • Company Paid HSA Contribution
  • Healthcare and Dependent Care Flexible Spending Accounts
  • 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
  • Back-up childcare and parenting support resources
  • Commuter benefits
  • Employee discounts and perks program

Interested in this job?

Jobs Related To Tesla Software Engineer, Generalist, AI Inference

Sr. Machine Learning Engineer, Autobidder

Senior Machine Learning Engineer position at Tesla focusing on developing forecasting algorithms for energy storage systems and renewable assets.

Software Engineer, Camera, AI Platforms

Senior Software Engineer position at Tesla focusing on camera systems and AI platforms for autonomous driving and robotics applications.

Software Engineer, Performance Modeling, Dojo

Senior Software Engineer position at Tesla, focusing on performance modeling and simulation for AI training architecture, offering competitive salary and comprehensive benefits.

Fullstack Software Engineer, Machine Learning Platform, AI Infrastructure

Senior Software Engineer role at Tesla working on ML infrastructure for autonomous vehicles and robotics, offering $116K-$360K salary plus benefits.

Sr. Machine Learning Engineer, Navigation, Tesla Bot

Senior Machine Learning Engineer position at Tesla focusing on navigation systems for Tesla Bot, involving deep learning and computer vision.