"jax" Jobs

5 open tech roles matching “jax”, taken straight from company career pages — not reposted from other job boards. Most in demand right now: JAX, PyTorch, AI/ML. Every listing is re-checked daily and closed roles are removed.

Showing 5 of 5 results

Research Engineer, Machine Learning (RL Velocity) Remote Visa

Anthropic Remote-Friendly (Travel-Required) | San Francisco, CA | New York City, NY $500k–$850k/yr Published 3 months ago

JAX AI/ML PyTorch

Apply →

Research Scientist, Multimodal Alignment, Safety, and Fairness

Google DeepMind Kirkland, Washington, US; Mountain View, California, US; New York City, New York, US $147k–$211k/yr Published 4 months ago

JAX multimodal learning reinforcement learning Python Gemax

Apply →

Product Manager II - Model Lab Remote

Datadog New York, New York, USA $155k–$190k/yr Published 11 months ago

JAX AI/ML PyTorch TensorFlow

Apply →

AI Applied Scientist Remote

Figma San Francisco, CA • New York, NY • United States $153k–$376k/yr Published 7 months ago

C++ JAX Machine Learning AI/ML Java

Apply →

Research Engineer, Machine Learning (Reinforcement Learning) Remote Visa

Anthropic San Francisco, CA | New York City, NY $500k–$850k/yr Published 1 year ago

C++ Trio JAX async/concurrent programming Rust

Apply →