[ engineering ]

Senior / Staff Ml Training Optimization Engineer

Waabi
Location
Remote
Work mode
Remote
Type
full time
Listed

Applications go directly to Waabi. Ramai takes no fee for placement and no money for ranking.

[ About this role ]

This is a Senior/Staff ML Training Optimization Engineer role at Waabi. The engineer will build standardized distributed training frameworks, profile model runtime and memory for performance bottlenecks, and evaluate emerging technologies for training and inference. They will also work with researchers on best practices for resource usage and create tooling and dashboards.

Skills and focus areas

  • distributed training
  • performance profiling
  • cuda kernel development
  • quantization-aware training
  • compilation techniques
  • deployment techniques
  • resource optimization
  • tooling development
  • dashboard creation
  • python
  • c++
  • rust

Summary written by Ramai from the original posting. Full details and requirements are on Waabi’s careers page.

[ More at Waabi ]

See all 92 open roles at Waabi →

[ Similar roles at other startups ]

Browse all open engineering roles →

[ Not quite right? ]

Tell the assistant what you’re after — stage, location, stack, visa sponsorship — and it searches every open role for you. A few questions are free, no account needed.