Software Engineer, Generalist, Ai Infrastructure

As a Software Engineer within Autopilot, you will work on reinforcing, optimizing, and scaling our neural network training infrastructure.

At the core of our self-driving capabilities, there are different neural networks that the Deep Learning team is designing to train large amounts of data. Robustly training jobs at scale, should it be for production models or quick experiments, and completing them in the shortest amount of time possible, is critical to our mission

• Practical experience programming in Python and/or C/C++
• Proficient in system-level software, in particular hardware-software interactions and resource utilization
• Understanding of modern machine learning concepts and state of the art deep learning
• Experience working with training frameworks, ideally PyTorch
• Demonstrated experience scaling neural network training jobs across clusters of GPU’s
• Experience programming in Cuda
• Profiling and optimizing CPU-GPU interactions (pipelining compute/transfers, etc.)
• Devops experience, in particular dealing with clusters of training nodes, and filesystems for very large amount of training data