Staff Software Engineer, ML Acceleration

stackav · Remote

Posted: 10 days ago
Last confirmed live: 1 day ago

What this role involves

Stack AV is hiring a Staff Software Engineer for ML Acceleration to optimize machine learning training systems. The role involves profiling models, implementing GPU optimizations with CUDA and Triton, and collaborating with ML researchers. Requires 5+ years experience with GPU programming and strong C++/Python skills.

Skills this posting asks for

c++
python
gpu programming
cuda
triton
pytorch
tensorrt
onnx
custom gpu kernel development
deep learning frameworks
autonomous vehicles

Requirements

5 years of experience
Level: senior

From the employer’s posting

About Stack: Stack is developing revolutionary AI and advanced autonomous systems designed to enhance safety, reliability, and efficiency of modern operations. Stack's autonomous technology incorporates cutting-edge advancements in artificial in…

Read the full description on stackav’s careers page

Apply without filling the form

Approve this role and the application is completed for you, including a résumé tailored to it. You get a confirmation when it lands, and a credit is only spent when a submission is confirmed.

Claim 25 free applications

Other roles at stackav

All 5 roles at stackav →