JAX jobs
Open developer roles that use JAX, pulled straight from company job boards and refreshed every few hours.
404
open roles
379
posted this week
156
fully remote
143
pay $150k+
Page 16
Filter in search →
Applied Machine Learning Engineer
Bjak - United States - Remote
4 days ago
Backend Developer
Bjak - United States - Remote
4 days ago
Performance Engineer (Inference, Training & GPU)
World Labs - San Francisco, CA - On-site
$200k–300k4 days ago
Machine Learning Scientist, Biomolecular Structure Prediction
Profluent - Emeryville, CA - On-site
4 days ago
Sr. Model Engineer, Post-Training & RL
SpaceXAI - Palo Alto, CA - On-site
7 days ago
Sr. Model Engineer, Mid-Training
SpaceXAI - Palo Alto, CA - On-site
7 days ago
Co-op, Agentic and Generative Protein Design
Lila Sciences - Cambridge, MA +1 - Hybrid
7 days ago
Research Engineer / Scientist, Pre-training
Anthropic - San Francisco, CA +2 - Hybrid
7 days ago
Senior Principal Model Bring-up Engineer
SambaNova - Austin, TX +1 - On-site
11 days ago
Research Scientist II – Deepfake Detection & Authentication (Video & Audio)
Pindrop - US - Remote; US-Remote - Remote
13 days ago
Sr. AI Researcher, Agentic Systems
OpenSpace - U.S. - Remote
13 days ago
Security Engineer / Research Engineer, RL Security
Anthropic - San Francisco, CA +2 - On-site
$180k–280k20 days ago
Staff / Principal Research Engineer, AI Safety Mitigations
Lila Sciences - Cambridge, MA +2 - Hybrid
$200k–320k24 days ago
2027 Summer Intern, PhD, Research, Post Training
Waymo - Mountain View, CA +1 - On-site
1 month ago
Manager, Applied AI Engineering (Megas)
Anthropic - San Francisco, CA +1 - Hybrid
$220k–300k1 month ago
Senior / Principal Scientist, Molecular Simulations
Flagship Pioneering, Inc. - Cambridge, MA - On-site
1 month ago
Scientific Machine Learning Engineer
Schrödinger - New York, NY - On-site
1 month ago
Staff+ Research Scientist, Multi-Agent
Anthropic - San Francisco, CA +2 - On-site
1 month ago
Senior ML Scientist, AI for Protein Engineering
Lila Sciences - San Francisco, CA - On-site
1 month ago
Machine Learning Scientist, AI for Protein Engineering
Lila Sciences - San Francisco, CA - On-site
1 month ago
AI Systems Researcher/Engineer
Brave - London, UK - Hybrid
2 months ago
Senior/ Staff AI Research Engineer
Xaira Therapeutics - Seattle, WA +1 - Hybrid
2 months ago
Staff Software Engineer, RL Environments
Scale AI - San Francisco, CA +1 - On-site
2 months ago
AI Scientist Intern, Computational Protein Design
Xaira Therapeutics - Seattle, WA +1 - On-site
2 months ago
AI in Residence, Computational Protein Design
Xaira Therapeutics - Seattle, WA - On-site
2 months ago