Senior Staff Software Engineer, ML Acceleration Lead
Stack is focused on developing AI and autonomous systems to improve safety, reliability, and efficiency in the trucking industry. They are seeking a Senior Staff Software Engineer to lead the ML Acceleration team.
Sign up for our
weekly newsletter
of fresh jobs
Skills
GPU programming
C++ and Python programming
ML optimization
CUDA programming
Deep learning frameworks
TensorRT implementation
Responsibilities
Analyze and profile ML models to identify performance bottlenecks.
Use OSS tools to enhance the platform for profiling models.
Automate the process of exporting models to optimized formats.
Implement optimizations using CUDA and Triton.
Collaborate with ML researchers on model accuracy and speed.
Lead cross-team projects related to model optimization.
Collaborate with teams to understand data requirements.
Education
Bachelor’s or Master’s degree in Computer Science, Engineering, or related field.
Benefits
Equal opportunity workplace
Diverse team culture
Commitment to inclusion and innovation
To read the complete job description, please click on the ‘Apply’ button