Software Engineer, ML Infra, Dojo

Tesla is seeking a Machine Learning Software Engineer to join their Dojo team, working on their custom-built supercomputer designed for AI training. This role sits at the crucial intersection of hardware acceleration and neural network development, specifically focusing on Tesla's Full Self-Driving (FSD) capability. You'll work alongside world-class ML Researchers, Compiler and Hardware Engineers to optimize and scale neural network training infrastructure.

The position offers an exciting opportunity to work on cutting-edge technology that directly impacts the future of autonomous driving. You'll be responsible for bridging the gap between Tesla's innovative Dojo training accelerator and the neural networks developed by the Autopilot ML team. This involves performance optimization, scaling solutions, and maintaining high-availability systems.

The ideal candidate should have a strong background in both software engineering and machine learning, with practical experience in Python, C++, and modern ML frameworks like PyTorch. System-level software expertise and understanding of hardware-software interactions are crucial for this role. You'll be working in Tesla's headquarters in the San Francisco Bay Area, contributing to one of the most advanced AI training systems in the industry.

Tesla offers a comprehensive benefits package including competitive salary, equity opportunities, and extensive health coverage. This is an excellent opportunity for someone passionate about pushing the boundaries of AI and hardware acceleration while working on technology that will revolutionize transportation.