Staff Software Engineer, ML Acceleration

Stack is developing revolutionary AI and advanced autonomous systems to enhance safety, reliability, and efficiency of modern operations. The ML Training Acceleration team aims to accelerate machine learning iterations and deliver a reliable, scalable, user-friendly, and observable training system.

Requirements

• Bachelor’s or Master’s degree in Computer Science, Engineering, or a related field
• 5+ years of experience (including experience with GPU programming and optimization)
• Strong programming skills in C++ and Python
• Proven experience in GPU programming and optimization
• Familiarity with deep learning frameworks, especially PyTorch
• Autonomous vehicles (AV) experience is a bonus

Benefits

• Equal opportunity workplace
• Diverse teams produce the best ideas and outcomes
• Inclusive culture
• Entrepreneurship and innovation