Description
Waabi is hiring a Distributed Training Engineer to build standardized distributed training frameworks for research and production, profile model runtime and memory, evaluate emerging training and inference technologies, and create tooling and dashboards. The role requires a computer science, robotics, or similar technical degree, at least four years of industry experience, proficiency in Python, C++, or Rust, and experience with PyTorch or Jax. The US yearly salary range is $141,000 to $249,000, with medical, dental, and vision coverage and work-from-home support.
