Skill
JAX AI jobs
Page 7 of 22 · 505 jax ai jobs
Open roles
Snowflake
Research Intern (AI) - 2026
PL-WarsawOn-site · Full-time · Intern
1 month ago
Aging
NVIDIA
Senior Software Engineer, ML
IN / Bengaluru, INOn-site · Full-time · Senior
2 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
2 months ago
May be filled
Mach9
Perception Intern, Summer 2026
San Francisco, CA, USOn-site · Internship · Intern
2 months ago
May be filled
Ideogram
Research Engineer / Research Scientist (Pre-training)
Toronto, NY, USOn-site · Full-time
2 months ago
May be filled
Ideogram
Research Engineer / Research Scientist (Post-training)
Toronto, NY, USOn-site · Full-time
2 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
2 months ago
May be filled
Mistral AI
Applied Scientist / Research Engineer
Seoul, KROn-site · Full-time
2 months ago
May be filled
Mistral AI
Senior/Staff Applied Scientist/Research Engineer
Seoul, KRHybrid · Full-time · Staff / Principal
2 months ago
May be filled
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
2 months ago
Aging
NVIDIA
CUDA Libraries and Frameworks Product Marketing Manager
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Manager / Lead
2 months ago
Aging
Skydio
PhD Autonomy Engineer Intern - Planning & Controls (Reinforcement Learning)
Zürich, CHHybrid · Internship · Intern
2 months ago
May be filled
Reality Defender
Staff AI Engineer (Audio)
— GlobalRemote · Full-time · Staff / Principal
$185K-$235K/yr
May be filled
Reality Defender
Staff ML Engineer (Audio)
— GlobalRemote · Full-time · Staff / Principal
$185K-$235K/yr
May be filled
Reality Defender
Applied AI Engineer II (Audio)
— GlobalRemote · Full-time
$160K-$210K/yr
May be filled
Lila Sciences
Co-Op, ML Scientist for Biology
San Francisco, CA, US / CA, USOn-site
2 months ago
May be filled
NVIDIA
Senior ML Compiler Engineer
Redmond, WA, US / US, WA, USOn-site · Full-time · Senior
2 months ago
Aging
NVIDIA
Senior Scientific Machine Learning Engineer – Earth-2
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
2 months ago
Aging
NVIDIA
Senior Data Center Performance Engineer - Benchmarking and Optimization
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
2 months ago
Aging