Senior Jax Jobs

42 open positions · Updated 5 months ago

Average salary (USD/year): 241.0k–464.4k/yr

42 roles Senior 42 3–10 yrs experience Bellevue, WA / Sunnyvale, CA +27 more Full-time newest 1 week ago

Also hiring: PyTorch Python TensorFlow C++ Kubernetes reinforcement learning

42 Jax roles across 25 companies, most in AI/ML; 2% fully remote; typical advertised salary $204.5k

Who is hiring (25 companies)

Anthropic (9) DeepL (4) Graphcore (4) Coreweave (2) Preference Model (2)

Role types

AI/ML (42)

Work arrangement: 1 fully remote · 16 hybrid · 8 on-site · 17 not stated

Advertised salaries: p25 $175.8k · median $204.5k · p75 $341.3k (from 26 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 2 of 42 positions

Search with filters →

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

Apply →

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

Apply →

← Prev Page 3 of 3