Staff Software Engineer, ML Training Infrastructure

Description: Stack is developing AI and autonomous systems for the trucking industry, focusing on safety, reliability, and efficiency. The ML Training team aims to enhance AV development velocity through a reliable and scalable training system. The Staff Software Engineer will design and develop a high-performance multi-tenant AI training platform, collaborating across various teams. Requirements: Bachelor’s or Master’s degree in Computer Science, Engineering, or related field. 6+ years of experience with ML Platforms and building ML-based applications. Strong programming skills in Python. Experience with Lance, PyTorch, Ray Data, or similar technologies. Proven ability to build scalable, reliable infrastructure in fast-paced environments. Deep understanding of design tradeoffs and ability to communicate them effectively. Experience with model training, optimization, or large-scale data processing pipelines. Autonomous vehicles (AV) experience is a bonus. Strong analytical, problem-solving, and communication skills. Benefits: Commitment to diversity and inclusion in the workplace. Opportunity to work on cutting-edge technology in the autonomous vehicle sector.