JAX jobs
Open developer roles that use JAX, pulled straight from company job boards and refreshed every few hours.
405
open roles
380
posted this week
156
fully remote
144
pay $150k+
Page 9
Filter in search →
Research Scientist/Engineer
Astera Institute - Emeryville, CA - On-site
$150k–300k4 days ago
Research Scientist: Hierarchical RL Agents
Astera Institute - Emeryville, CA - On-site
$150k–300k4 days ago
Research Scientist - Simplex
Astera Institute - Emeryville, CA - On-site
$140k–200k4 days ago
Machine Learning Researcher - Springtail
Astera Institute - Emeryville, CA - Hybrid
$150k–300k4 days ago
ML Infra Engineer, Modeling
Physical Intelligence - San Francisco, CA - On-site
4 days ago
Member of Technical Staff, Research Engineer
Runway - Remote - Remote
$270k–370k4 days ago
Research Engineer, Data Foundations
Runway - Remote - Remote
$270k–370k4 days ago
Machine Learning Engineer, Speech LLM Training - San Francisco
Plaud - San Francisco, CA - Hybrid
$195k–365k4 days ago
Applied AI Engineer
Twenty - Arlington, VA - On-site
$159k–263k4 days ago
AI research scientist
WRITER - San Francisco, CA +1 - Hybrid
$199k–325k4 days ago
AI engineer
WRITER - San Francisco, CA +1 - Hybrid
$198k–259k4 days ago
Machine Learning Engineer, Applied AI Co-Op Intern, Winter 2027
Brain Co. - San Francisco, CA - On-site
$96k–120k4 days ago
Machine Learning Engineer, Applied AI Intern (Summer 2027)
Brain Co. - San Francisco, CA - On-site
$96k–120k4 days ago
AI/ML Engineer, 2027 New Grad
Brain Co. - San Francisco, CA - On-site
$140k–190k4 days ago
Member of Technical Staff — RL & Agentic Model Training
Embedding VC - San Francisco, CA - On-site
4 days ago
Member of Technical Staff — Post-Training Infrastructure
Embedding VC - San Francisco, CA - On-site
4 days ago
Member of Technical Staff — Agent Post-Training
Embedding VC - San Francisco, CA - On-site
4 days ago
AI Research Scientist
Embedding VC - Milpitas, CA - On-site
4 days ago
2027 Internship Onboard Infrastructure Engineer, ML Inference
Bedrock Robotics Inc - San Francisco, CA - On-site
4 days ago
Machine Learning Engineer: Perception
Bedrock Robotics Inc - San Francisco, CA - On-site
4 days ago
Machine Learning Engineer: Imitation and Reinforcement Learning for Robotics
Bedrock Robotics Inc - San Francisco, CA - On-site
4 days ago
Forward Deployed Engineer (Training)
Baseten - San Francisco, CA +1 - On-site
$200k–400k4 days ago
Principal Applied Scientist - AI
UiPath - Bellevue, WA - On-site
$218k–265k4 days ago
AI / ML Engineer - Known
Pear VC - San Francisco, CA - On-site
4 days ago
Machine Learning Engineer, Driving Foundation Models
42dot - Sunnyvale, CA - On-site
$175k–250k4 days ago