Software Engineer, Generalist, AI Inference
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 their deployment on compute-constrained edge devices, including CPUs, GPUs, and Tesla's custom AI ASIC.
The role offers an exciting opportunity to work on state-of-the-art AI systems that directly impact Tesla's autonomous driving and robotics initiatives. You'll be responsible for building robust AI frameworks and infrastructure, optimizing neural network performance, and collaborating with AI scientists and compiler engineers to enhance model efficiency.
The ideal candidate should have strong expertise in Python, modern C++, and machine learning frameworks like PyTorch. Experience with CUDA and deep understanding of computer architecture is essential. This position offers competitive compensation ranging from $104,000 to $360,000 annually, plus additional cash and stock awards.
Tesla provides comprehensive benefits including premium healthcare options with $0 payroll deductions, 401(k) with employer match, stock purchase plans, and various family-friendly benefits. The company's commitment to innovation, sustainability, and cutting-edge technology makes it an exciting place for engineers looking to make a significant impact in the automotive and AI industries.
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
- 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 with $0 payroll deduction options
- Family-building, fertility, adoption and surrogacy benefits
- Dental and vision plans
- Health Savings Account (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
Jobs Related To Tesla Software Engineer, Generalist, AI Inference
