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

Showing 3 of 3 results

Applied AI/ML Engineer (Agents) Hybrid

Join CuspAI as an Applied AI/ML Engineer to design intelligent agents for autonomous materials discovery.

CuspAI Amsterdam, NL Published 3 months ago

Active Learning JAX Bayesian optimization PyTorch

Flexible on stack

Apply →

Platform Engineer

Join CuspAI as a Platform Engineer to build and maintain ML infrastructure that empowers researchers in materials science.

CuspAI London, UK Published 3 weeks ago

Terraform Kapitan JAX Go PyTorch

Flexible on stack

Apply →

AI Research Engineer Hybrid

Join Graphcore as an AI Research Engineer to advance AI research and collaborate on next-gen AI hardware.

Graphcore Cambridge, UK Published 4 months ago

C++ Triton JAX CUDA PyTorch

Flexible on stack

Apply →