Jax Jobs at Anthropic

12 open positions · Updated 4 months ago

Average salary: 350.4k–733.8k/yr

12 of 550 open roles Senior 10, Mid 2 5–10 yrs experience London, UK +5 more Full-time newest 4 months ago

Also hiring: PyTorch LLM AI/ML Kubernetes Python reinforcement learning

Showing 12 of 12 positions

Search with filters →

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 →

Performance Engineer, GPU Remote Visa

Anthropic San Francisco, CA | New York City, NY | Seattle, WA $280k–$850k/yr Published 11 months ago

FP8 NVLink Triton JAX CUTLASS

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, Visual Knowledge Work Remote Visa

Anthropic New York City, NY; San Francisco, CA; Seattle, WA $350k–$850k/yr Published 7 months ago

JAX Large Language Models Machine Learning AI/ML reinforcement learning

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 →

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 →

Staff+ Software Engineer, Safeguards ML Infrastructure Remote Visa

Anthropic San Francisco, CA $320k–$405k/yr Published 1 year ago

AWS Spark JAX Large Language Models transformer architectures

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, Interpretability Remote Visa

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

JAX CUDA Interpretability Java Go

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 →