Skill
JAX AI jobs
JAX AI jobs: 572 live roles on Kairos, the most recent posted 1 hour ago. Hiring now: NVIDIA, Clera, Allen Institute for AI, Institute of Foundation Models, Cerebras.
Updated 1 hour ago
Open roles
View all 572 roles
Hark
Member of Technical Staff, LLM & Production
San Jose, CA, USOn-site
1 hour ago
NVIDIA
Software Engineer, DL Kernel Libraries - New College Grad 2027
Santa Clara, CA, US / US, CA, USOn-site · Full-time
1 day ago
NVIDIA
Senior Deep Learning Software Engineer, Inference and Model Optimization
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
3 days ago
Cognition
Research Engineer, Safety & Alignment
San Francisco, CA, USOn-site · Full-time
3 days ago
NVIDIA
Robotics Simulation and Evaluation PhD Intern - Summer 2027
Zurich / CHOn-site · Full-time · Intern
3 days ago
NVIDIA
PhD Research Intern, Robotics - Summer 2027
Zurich / CHOn-site · Full-time · Intern
3 days ago
GenBio AI
Research Scientist Intern, AI Molecular Design
Palo Alto, CA, USOn-site · Internship · Intern
$8K-$10K/month
Mach9
Perception Intern, Fall 2026
San Francisco, CA, USOn-site · Internship · Intern
4 days ago
Clera
Research Engineer
Zürich, CHOn-site · Full-time
4 days ago
Zoox
Engineering Manager, ML Performance Optimization
Foster City, CA, USHybrid · Full-time · Manager / Lead
$276K-$343K/yr
Institute of Foundation Models
Machine Learning Engineer — Reinforcement Learning
Sunnyvale, CA, USOn-site · Full-time
$150K-$450K/yr
Institute of Foundation Models
Machine Learning Engineer — Pre-training (LLM)
Sunnyvale, CA, USOn-site · Full-time
$150K-$450K/yr
NVIDIA
Performance Engineer - Deep Learning
Santa Clara, CA, US / US, CA, USOn-site · Full-time
1 week ago
Clera
Founding Reinforcement Learning Engineer
San Francisco, CA, USOn-site · Full-time
1 week ago
Clera
Research Engineer
Zürich, CHOn-site · Full-time
1 week ago
Medra
Research Engineer, Post-training
San Francisco, CA, USOn-site · Full-time
1 week ago
NVIDIA
Software Engineer Intern, AI and DL Kernel Libraries - 2027
Shanghai, CN / CNOn-site · Full-time · Intern
1 week ago
Anthropic
Research Engineer / Research Scientist, RL Frontiers
San Francisco, CA, US / Seattle, WA, USOn-site
$500K-$850K/yr
Aging
PostEra
Machine Learning Researcher - Agentic Science
Remote · Full-time
1 week ago
Cerebras
ML Runtime and Kernel Engineer - Core ML
Sunnyvale, CA, US / Toronto, ON, CAHybrid · Full-time
2 weeks ago
Cerebras
ML Algorithm Mapping and Performance Engineer, Core ML
Sunnyvale, CA, USHybrid · Full-time
2 weeks ago
Allen Institute for AI
Research Engineer, Climate Modeling
Seattle, WA, USOn-site
2 weeks ago
Aging
Allen Institute for AI
Research Scientist, Climate Modeling
Seattle, WA, USOn-site
2 weeks ago
Aging
Allen Institute for AI
Young Investigator, Climate Modeling
Seattle, WA, USOn-site
2 weeks ago
Aging