"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, Python, reinforcement learning. Every listing is re-checked daily and closed roles are removed.

Showing 5 of 5 results

Research Engineer, Materials Science

Join Google DeepMind as a Research Engineer to advance materials science through AI and computational simulations.

Google DeepMind Mountain View, California, US $141k–$202k/yr Published 9 months ago

C++ molecular dynamics Diffusion Models Transformers Python

Flexible on stack

Apply →

Research Scientist, Gemini Safety

Google DeepMind Mountain View, California, US Published 4 months ago

JAX AI/ML reinforcement learning

Apply →

Research Engineer, Multimodal Reasoning For Information Literacy

Google DeepMind Mountain View, California, US $174k–$252k/yr Published 10 months ago

JAX Machine Learning PyTorch TensorFlow Python

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 →

Research Engineer, Human Understanding

Google DeepMind Los Angeles, California, US; Mountain View, California, US $174k–$252k/yr Published 4 months ago

JAX multimodal learning reinforcement learning Python Generative AI

Apply →