Skill
JAX AI jobs
Page 10 of 24 · 575 jax ai jobs
Open roles
NVIDIA
Applied Research Intern, NLP - Fall 2026
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Intern
3 months ago
Aging
NVIDIA
Senior Manager, Software Engineering - JAX
DERemote · Full-time · Manager / Lead
3 months ago
Aging
AMI
AMI Engineer - Singapore
Paris, FROn-site · Full-time
3 months ago
May be filled
AMI
AMI Engineer - Paris
Paris, FROn-site · Full-time
3 months ago
May be filled
Lila Sciences
Co-Op, LS AI, ML Scientist for Protein Engineering
San Francisco, CA, US / CA, USOn-site
3 months ago
May be filled
Snowflake
Research Intern (AI) - 2026
PL-WarsawOn-site · Full-time · Intern
3 months ago
May be filled
NVIDIA
Senior Software Engineer, ML
IN / Bengaluru, INOn-site · Full-time · Senior
3 months ago
Aging
Field AI
Senior Machine Learning Engineer
Seattle, WA, US / Irvine, CA, USOn-site · Full-time · Senior
$180K-$215K/yr
May be filled
Figure AI
Staff AI Inference and Acceleration Engineer
San Jose, CA, USOn-site · Staff / Principal
3 months ago
May be filled
Mach9
Perception Intern, Summer 2026
San Francisco, CA, USOn-site · Internship · Intern
3 months ago
May be filled
Ideogram
Research Engineer / Research Scientist (Pre-training)
Toronto, NY, USOn-site · Full-time
3 months ago
May be filled
Ideogram
Research Engineer / Research Scientist (Post-training)
Toronto, NY, USOn-site · Full-time
3 months ago
May be filled
Cosine
ML Systems Engineer - Model Training and Infrastructure (SWE-focused LLMs)
London, GBOn-site · Full-time
GBP 80K-GBP 110K/yr
May be filled
Runway
Member of Technical Staff, Research Engineer (GPU Performance)
Remote · Full-time
$270K-$370K/yr
May be filled
Moonlake AI
Member of Technical Staff - Code Generation
San Francisco, CA, USOn-site · Full-time
3 months ago
May be filled
Mistral AI
Applied Scientist / Research Engineer
Seoul, KROn-site · Full-time
3 months ago
Aging
Mistral AI
Senior/Staff Applied Scientist/Research Engineer
Seoul, KRHybrid · Full-time · Staff / Principal
3 months ago
Aging
Asari AI
Member of Technical Staff - Kernel Engineer
San Francisco, CA, USOn-site · Full-time
$150K-$250K/yr
May be filled
Reality Defender
Audio ML Engineer II
— GlobalRemote · Full-time
$160K-$210K/yr
May be filled
CuspAI
Applied AI/ML Engineer (Agents/RL)
London, GB / Cambridge, GBHybrid · Full-time
3 months ago
Aging
NVIDIA
CUDA Libraries and Frameworks Product Marketing Manager
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Manager / Lead
3 months ago
Aging
Lila Sciences
Co-Op, ML Scientist for Biology
San Francisco, CA, US / CA, USOn-site
3 months ago
May be filled
NVIDIA
Senior ML Compiler Engineer
Redmond, WA, US / US, WA, USOn-site · Full-time · Senior
3 months ago
Aging
NVIDIA
Senior Scientific Machine Learning Engineer – Earth-2
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
3 months ago
Aging