Waabi Posted 30+ days ago

Senior / Staff ML Training Optimization Engineer

Location

San Francisco

Hybrid

Contract

Permanent

Full-time

Salary

$141–249k

Gross per year

Experience

4+ years

Lead level

What they're looking for

MS/PhD or Bachelors degree with a minimum of 4 years of industry experience in Computer Science, Robotics and/or similar technical field(s) of study.
Solid coding proficiency in a variety of coding languages including Python, C++ or Rust.
Experience in deep learning frameworks such as PyTorch or Jax.
Skilled in profiling CPU and GPU code using tools such as PyTorch Profiler and NVIDIA Nsight.
Open-minded and collaborative team player with willingness to help others.