ML Platform Engineer

Stack AV is at the forefront of revolutionizing the trucking transportation industry through advanced autonomous systems and AI technology. As part of the ML Platform organization, we're seeking a Senior ML Platform Engineer to join our training and deployment team. This role is crucial in building and optimizing the platform that enables our AI team to develop, test, and deploy models on autonomous vehicles.

The ideal candidate will bring deep expertise in GPU programming and optimization, with a proven track record of implementing high-performance ML solutions. You'll work with cutting-edge technologies including CUDA, Triton, and custom kernels, while collaborating with ML researchers to balance model accuracy and speed.

This position offers an opportunity to work on challenging problems in autonomous vehicle technology, utilizing the latest advancements in ML inference and accelerators. You'll be responsible for developing efficient solutions for model export, optimization, and profiling across various hardware platforms.

We're looking for someone who can not only maintain high engineering standards but also foster a culture of excellence within the team. The role requires strong analytical skills and the ability to communicate complex technical concepts effectively.

Join our diverse and inclusive team where innovation and entrepreneurship are valued across all backgrounds. Note that this position may be subject to U.S. national security requirements and export control regulations, requiring verification of citizenship status and potential export licenses.