"jax" Jobs

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

Showing 9 of 9 results

Research Engineer, Interpretability Remote Visa

Anthropic San Francisco, CA $315k–$560k/yr Published 9 months ago

JAX CUDA Interpretability Java Go

Apply →

Research Engineer, Machine Learning (RL Velocity) Remote Visa

Anthropic London, UK £370k–£630k/yr Published 4 months ago

JAX AI/ML PyTorch

Apply →

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 Engineer, Pretraining Scaling - London Visa

Anthropic London, UK £260k–£630k/yr Published 10 months ago

JAX large-scale distributed systems PyTorch LLM evaluation infrastructure

Apply →

Research Engineer, Pretraining Scaling Visa

Anthropic San Francisco, CA $350k–$850k/yr Published 10 months ago

ML frameworks JAX model operations AI/ML large-scale distributed systems

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 →

Research Engineer, Machine Learning (Reinforcement Learning) Remote Visa

Anthropic London, UK £260k–£630k/yr Published 6 months ago

C++ Trio JAX async/concurrent programming Rust

Apply →

Research Engineer, Discovery Remote Visa

Anthropic San Francisco, CA $350k–$850k/yr Published 1 year ago

Beam AWS Spark JAX AI/ML

Apply →

Research Engineer, Performance RL (Reinforcement Learning) Remote Visa

Anthropic San Francisco, CA $350k–$850k/yr Published 5 months ago

Triton JAX CUDA reinforcement learning PyTorch

Apply →