JAX Jobs

Explore JAX roles on Inference Jobs and apply today.

2mo agoXA

Member of Technical Staff, Midtraining

xAI

Palo Alto, California, United States (On-site)$180K – $440K Yearly
3mo agoCO

Senior Member of Technical Staff, Multimodal AI

Cohere

San Francisco, California, United States or Remote (Worldwide)
2w agoAN

Research Engineer, Pretraining Scaling (London)

Anthropic

London, England, United Kingdom (On-site)£260K – £630K Yearly
3mo agoAN

Research Engineer, Audio

Anthropic

San Francisco, California, United States (Hybrid)$350K – $500K Yearly
2w agoGR

Senior Machine Learning Engineer (Large Systems)

Graphcore

Gdańsk, Pomeranian Voivodeship, Poland (On-site)zł 260.4K – zł 352.2K Yearly
3w agoTM

Research Engineer, Infrastructure, Training Systems

Thinking Machines Lab

San Francisco, California, United States (On-site)$350K – $475K Yearly
3mo agoCO

Member of Technical Staff, Modeling

Cohere

London, England, United Kingdom or Remote (Worldwide)
3w agoTM

Research Engineer, Infrastructure, Tinker

Thinking Machines Lab

San Francisco, California, United States (On-site)$350K – $475K Yearly
2w agoGR

Research Engineer

Graphcore

Bristol, England, United Kingdom (On-site)