Staff Software Engineer, ML Acceleration

Stack AV

Remote regions

US

Benefits

Similar Jobs

See all

Responsibilities:

  • Analyze ML models to identify and resolve performance bottlenecks.
  • Incorporate OSS tools to enable ML engineers self-sufficiently profile and optimize models.
  • Deliver solutions to streamline model deployment across various hardware platforms.

Qualifications:

  • Strong programming skills in C++ and Python
  • Proven experience in GPU programming and optimization
  • Familiarity with deep learning frameworks, especially PyTorch

Technical Skills:

  • CUDA programming
  • Triton language for GPU kernels
  • PyTorch optimization techniques

Stack AV

Stack is developing revolutionary AI and advanced autonomous systems designed to enhance safety, reliability, and efficiency of modern operations. With decades of experience creating and deploying real world systems for demanding environments, the Stack team is dedicated to developing an autonomous solution ecosystem tailored to the trucking industry's unique demands.

Apply for This Position