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.