flax

Recent positions

• Software Engineer

  waymo · Mar 31, 2026

  $240k – $259k

  mid · on-site · full time · Mountain View

  c++ python sql pytorch tensorflow jax +3
  Apply

• Research Scientist, Reinforcement Learning

  deepmind · Mar 14, 2026

  senior · on-site · London

  pytorch jax flax
  Apply

• Research Scientist, Multimodal Alignment, Safety, and Fairness

  deepmind · Mar 9, 2026

  $147k – $211k

  senior · on-site · full time · Kirkland, Mountain View, New York City

  python jax flax gemax
  Apply

• Senior Machine Learning Engineer, Driver Understanding and Evaluation

  waymo · Dec 17, 2025

  $204k – $259k

  senior · on-site · full time · Mountain View

  c++ python pytorch robotics tensorflow jax +4
  Apply

• Machine Learning Engineer, Driver Understanding and Evaluation

  waymo · Dec 17, 2025

  $170k – $216k

  mid · hybrid · full time · Mountain View

  c++ python pytorch tensorflow jax flax
  Apply

• Machine Learning Engineer, Simulation Realism

  waymo · Mar 13, 2025

  $213k – $263k

  senior · hybrid · full time · Mountain View

  c++ python pytorch tensorflow jax flax
  Apply

• Senior/Staff ML Engineer, 3D/4D World Modeling, Simulation

  waymo · Mar 6, 2025

  $213k – $263k

  senior · on-site · full time · Mountain View

  c++ python pytorch tensorflow jax gpu +2
  Apply