Description:

Stack is focused on AI advancements across diverse technical domains, transforming how AI is applied in the physical realm.
The training and deployment team is part of the ML Platform org at Stack AV, responsible for building, optimizing, testing, and deploying models on autonomous vehicles.
The Senior Staff Software Engineer, ML Acceleration Lead will design the architecture and lead a team to automate the optimization and deployment of complex ML models for next-gen AI Autonomous Vehicle applications.
The role requires a deep understanding of GPUs and optimization, excellent leadership skills, and the ability to drive technical excellence.
Responsibilities include analyzing and profiling ML models, enhancing the platform with OSS tooling, automating model export to optimized formats, implementing optimizations using CUDA and Triton, and collaborating with ML researchers.

Requirements:

A Bachelor’s or Master’s degree in Computer Science, Engineering, or a related field is required.
A minimum of 5 years of experience in GPU programming and optimization is necessary.
Strong programming skills in C++ and Python are essential.
Proven experience in GPU programming and optimization is required.
Familiarity with deep learning frameworks, especially PyTorch, is necessary.
Experience with CUDA programming, Triton language for GPU kernels, PyTorch optimization techniques, TensorRT implementation, ONNX model conversion and deployment, and custom GPU kernel development is required.
A deep understanding of GPU architectures and performance optimization is essential.
Proven ability to lead and mentor a team, manage projects, and drive technical initiatives is required.
Strong analytical and problem-solving skills are necessary.
Excellent verbal and written communication skills are required, with the ability to convey complex technical concepts to non-technical stakeholders.

Stack is committed to being an equal opportunity workplace, promoting diversity and inclusion across various dimensions.
The company fosters a culture of entrepreneurship and innovation.
Employees will have the opportunity to work on cutting-edge AI technology in the autonomous vehicle sector.
The position may involve remote work flexibility, allowing for a better work-life balance.
Stack AV complies with applicable U.S. national security laws, which may provide a structured and secure work environment.