Software Engineer, Generalist, AI Inference

Tesla is seeking a Software Engineer specializing in AI Inference to join their cutting-edge team focused on productionalizing ML models. This role sits at the intersection of machine learning and systems engineering, where you'll work on training and deploying large neural networks for efficient inference on compute-constrained edge devices. You'll be directly involved with Tesla's flagship products - Autopilot and Optimus - building the ML frameworks and infrastructure that power their AI capabilities.

The position offers an exciting opportunity to work with state-of-the-art technology, including Tesla's in-house AI ASIC, and collaborate with AI scientists and compiler engineers. You'll be responsible for optimizing network performance while minimizing latency, implementing custom GPU kernels, and building robust AI infrastructure for large-scale model training.

The ideal candidate should have strong programming skills in Python and C++, deep understanding of ML frameworks like PyTorch, and experience with CUDA programming. This role offers competitive compensation ranging from $104,000 to $360,000 annually, plus additional cash and stock awards. Tesla provides comprehensive benefits including medical coverage, 401(k) matching, and various family-friendly policies.

Working at Tesla means being at the forefront of AI innovation in automotive and robotics, with the opportunity to impact the future of autonomous driving and robotics technology. The company's commitment to diversity, equal opportunity, and reasonable accommodations ensures an inclusive work environment for all employees.