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 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.
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
- 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
Jobs Related To Tesla Software Engineer, Generalist, AI Inference
