Skill
JAX AI jobs
JAX AI jobs: 333 live roles on Kairos, the most recent posted 1 day ago. Hiring now: Snowflake, Physical Intelligence, Mach9, Ideogram, Lila Sciences.
Updated 23 hours ago
Open roles
View all 333 roles
Lila Sciences
Co-Op, LS AI, ML Scientist for Protein Engineering
San Francisco, CA, US / CA, USOn-site
1 day ago
Snowflake
Research Intern (AI) - 2026
PL-WarsawOn-site · Full-time · Intern
1 day ago
Snowflake
Post-Doctoral Researcher (Fixed-Term)
US-WA-BellevueOn-site · Full-time
$160K-$220K
Snowflake
AI System Research and Development Engineer - Optimization
US-WA-BellevueOn-site · Full-time
$200K-$265K
Snowflake
AI Research Scientist, New Grad – Agents & Reinforcement Learning
US-WA-BellevueOn-site · Full-time · Entry / Junior
$176K-$230K
NVIDIA
Senior Software Engineer, ML
IN / Bengaluru, INOn-site · Full-time · Senior
1 day ago
Field AI
Senior Machine Learning Engineer
Seattle, WA, US / Irvine, CA, USOn-site · Full-time · Senior
$180K-$215K/yr
CuspAI
Applied AI/ML Engineer (Agents/RL)
London, GB / Cambridge, GBHybrid · Full-time
3 days ago
Figure AI
Staff AI Inference and Acceleration Engineer
San Jose, CA, USOn-site · Staff / Principal
4 days ago
Spellbrush
AI Anime Researcher
San Francisco, CA, US / Tokyo, JPOn-site · Full-time
4 days ago
Rogo
AI Researcher
San Francisco, CA, US / New York City, NY, USOn-site · Full-time
4 days ago
Reflection AI
Member of Technical Staff - Research Software Engineer
San Francisco, CA, US / NY, USOn-site · Full-time
4 days ago
Physical Intelligence
ML Infra Engineer (TPU/Jax/Optimization)
San Francisco, CA, USOn-site · Full-time
4 days ago
Physical Intelligence
ML Infra Engineer
San Francisco, CA, USOn-site · Full-time
4 days ago
Netic
Applied AI Research Engineer
San Francisco, CA, USOn-site · Full-time
4 days ago
Mach9
Perception Intern, Summer 2026
San Francisco, CA, USOn-site · Internship · Intern
4 days ago
Mach9
ML Engineer
San Francisco, CA, USOn-site · Full-time
4 days ago
Lovable
Researcher, Post Training
London, GB / Stockholm, SEOn-site · Full-time
4 days ago
LanceDB
Senior Software Engineer
HQRemote · Full-time · Senior
$180K-$250K
Krea
ML Engineer - Personalization & Recommendation Systems
San Francisco, CA, USOn-site · Full-time
4 days ago
Ideogram
Research Engineer / Research Scientist (Pre-training)
Toronto, NY, USOn-site · Full-time
4 days ago
Ideogram
Research Engineer / Research Scientist (Post-training)
Toronto, NY, USOn-site · Full-time
4 days ago
Cosine
ML Systems Engineer - Model Training and Infrastructure (SWE-focused LLMs)
London, GBOn-site · Full-time
GBP 80K-GBP 110K
Applied Intuition
ML Runtime Optimization Engineer
Sunnyvale, CA, USOn-site · Full-time
4 days ago