JAX jobs
Open developer roles that use JAX, pulled straight from company job boards and refreshed every few hours.
406
open roles
383
posted this week
156
fully remote
145
pay $150k+
Page 2
Filter in search →
Principal Architect, AI/ML
Zencore - Remote - Remote
3 days ago
Human Interactive Driving Research Intern, World Models
Toyota Research Institute - Los Altos, CA - Hybrid
$94k–135k3 days ago
Human Interactive Driving Research Intern, World Models
Toyota Research Institute - Los Altos, CA - Hybrid
$94k–135k3 days ago
Human Interactive Driving Research Intern, Perception for Driving at the Limits
Toyota Research Institute - Los Altos, CA - Hybrid
$94k–135k3 days ago
Human Interactive Driving Research Intern, Decision-Making for Driving at the Limits
Toyota Research Institute - Los Altos, CA - Hybrid
$94k–135k3 days ago
Machine Learning Engineer - Advanced AI and Cognitive Systems
HRL Laboratories - Calabasas, CA - On-site
$141k–176k3 days ago
Research Scientist – World Modeling, Data
Institute of Foundation Models - Sunnyvale, CA - On-site
$150k–400k3 days ago
Research Scientist - Reinforcement Learning
Institute of Foundation Models - Sunnyvale, CA - On-site
$150k–450k3 days ago
Machine Learning Engineer — Reinforcement Learning
Institute of Foundation Models - Sunnyvale, CA - On-site
$150k–450k3 days ago
Machine Learning Engineer — Pre-training (LLM)
Institute of Foundation Models - Sunnyvale, CA - On-site
$150k–450k3 days ago
Data & Eval Operations Program Manager
Institute of Foundation Models - Sunnyvale, CA - On-site
$250k–450k3 days ago
AI Research Internship - WM
Institute of Foundation Models - Sunnyvale, CA - On-site
$100k–140k3 days ago
AI Research Internship - LLM
Institute of Foundation Models - Sunnyvale, CA - On-site
$100k–140k3 days ago
Senior Machine Learning Engineer, Perception
PlusAI - Santa Clara, CA - On-site
$150k–200k3 days ago
Research Scientist, Robotics
Tutor Intelligence - San Francisco, CA +1 - On-site
$180k–420k3 days ago
Research Scientist, Robotics
Tutor Intelligence - Watertown, MA - On-site
$180k–420k3 days ago
Senior / Staff Software Engineer, ML Datasets & Data Pipelines
Waabi - Toronto, ON +1 - Hybrid
$148k–260k3 days ago
Senior / Staff ML Training Optimization Engineer
Waabi - Dallas, TX +4 - Hybrid
$184k–272k3 days ago
Research Scientist, World Models
Waabi - Toronto, ON +3 - Hybrid
$155k–269k3 days ago
Research Scientist, Simulation Agents
Waabi - Dallas, TX +4 - Hybrid
$158k–269k3 days ago
Research Scientist, Neural Reconstruction
Waabi - Toronto, ON +3 - Hybrid
$155k–269k3 days ago
Research Engineer, World Models
Waabi - Toronto, ON +3 - Hybrid
$155k–269k3 days ago
Lead Product Manager, Hardware Platform
Waabi - San Francisco, CA +2 - Hybrid
$190k–270k3 days ago
Lead Product Manager, Autonomy Capabilities
Waabi - Pittsburgh, PA +3 - Hybrid
$165k–260k3 days ago
Senior Staff Machine Learning Engineer - Content Platform
Spotify - New York, NY - Hybrid
$281k–402k3 days ago