joblaze

Direct from source · No middlemen

Jax Jobs

112 open positions · Updated 5 months ago

Average salary (USD/year): 220.1k–408.6k/yr

112 roles Senior 42, Mid 39, Staff 17, Intern 8, Junior 2, Lead 2, C-level 1 1–10 yrs experience Amsterdam, NL +53 more Full-time, Internship newest 1 day ago

Also hiring: PyTorch Python TensorFlow C++ Kubernetes Machine Learning

112 Jax roles across 46 companies, most in AI/ML; 5% fully remote; typical advertised salary $200k

Who is hiring (46 companies)

Anthropic (12) Skild AI (12) Graphcore (9) DeepL (5) Perplexity (5)

Role types

AI/ML (110) Data Science (1) Other (1)

Work arrangement: 6 fully remote · 31 hybrid · 25 on-site · 50 not stated

Advertised salaries: p25 $152.3k · median $200k · p75 $277.6k (from 52 disclosed annual salaries, USD)

Counts are open roles Joblaze currently tracks on company career pages for this exact skill/location; salaries are advertised minimums, annual, converted to USD.

Showing 12 of 112 positions

Search with filters →

Internship - Search Machine Learning Engineer

Perplexity Belgrade Published 8 months ago

JAX Rust PyTorch TensorFlow Python

AI Applied Scientist Hybrid

Figma San Francisco, CA • New York, NY • United States $153k–$376k/yr Published 8 months ago

C++ JAX Machine Learning AI/ML Java

AI Research Engineer - Datadog AI Research (DAIR) Hybrid

Datadog Paris, France Published 1 year ago

C++ Ray JAX CUDA DeepSpeed

Research Engineer, Performance RL (Reinforcement Learning) Hybrid Visa

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

Triton JAX CUDA reinforcement learning PyTorch

Research Engineer, Machine Learning (Reinforcement Learning) Hybrid Visa

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

C++ Trio JAX async/concurrent programming Rust

Research Engineer, Machine Learning (Reinforcement Learning) Hybrid Visa

Anthropic San Francisco, CA | New York City, NY $500k–$850k/yr Published 1 year ago

C++ Trio JAX async/concurrent programming Rust

Research Engineer, Pretraining Scaling - London Visa

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

JAX large-scale distributed systems PyTorch LLM evaluation infrastructure

Research Engineer, Visual Knowledge Work Hybrid 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

Performance Engineer, GPU Hybrid Visa

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

FP8 NVLink Triton JAX CUTLASS

Research Engineer, Interpretability Hybrid Visa

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

JAX CUDA Interpretability Java Go

Research Engineer, Pretraining Scaling Visa

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

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

Research Engineer, Discovery Hybrid Visa

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

Beam AWS Spark JAX AI/ML

← Prev Page 6 of 6

Related searches

jax Remote jax in London jax in New York jax in San Francisco jax in Berlin jax in Amsterdam Python Ruby On Rails React Devops Machine Learning Golang