"jax" Jobs

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

Showing 3 of 3 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 4 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 5 months ago

JAX multimodal learning reinforcement learning Python Gemax

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 →