Senior Software Engineer, Training Efficiency

Highlights

• Improve runtime efficiency of input pipelines for large-scale training workloads.
• Build Perception and Planning models powering the world's most trusted driver.
• Develop ML solutions in close collaboration with teams at Google.

What they offer

• Discretionary annual bonus program
• Equity incentive plan
• Generous Company benefits program

Ideal candidate

A systems-minded engineer passionate about ML efficiency who thrives collaborating with researchers to unblock large-scale training and scale autonomous driving models.

Requirements

• B.S. in Computer Science, Math, or 5+ years equivalent real-world experience
• Proficient in distributed systems design with an understanding of ML data pipeline optimization
• Experience with ML frameworks, including TensorFlow and JAX
• Hands-on experience with libraries like Grain or tf.data service
• Solid programming skills in Python and C++
• Practical familiarity with profiling tools to uncover performance bottlenecks
• MS in Computer Science, Math preferred
• Familiarity with distributed dataflow frameworks like ML Pathways