Skill
JAX AI jobs
Page 3 of 22 · 505 jax ai jobs
Open roles
Clera
Founding Machine Learning Engineer
Remote · Full-time
3 weeks ago
Aging
Dyna Robotics
Research Engineer/Scientist, Simulation
Redwood City, CA, USOn-site · Full-time
3 weeks ago
Aging
NVIDIA
Research Scientist, AI for Clinical Translatability
US, NY, USRemote · Full-time
3 weeks ago
Lila Sciences
Staff Research Engineer, Scientific Computing and ML/Physics Infrastructure
San Francisco, CA, US / London, GBOn-site · Staff / Principal
$224K-$294K/yr
Aging
Lila Sciences
Scientist II / Senior ML Scientist, Data-Efficient Learning for Drug Discovery
San Francisco, CA, US / London, GBOn-site · Senior
$228K-$358K/yr
Aging
Lila Sciences
Scientist II / Senior ML Scientist, Cofolding and Structure-Aware ML
San Francisco, CA, US / London, GBOn-site · Senior
$228K-$358K/yr
Aging
NVIDIA
Global Account Manager, Developer Relations - Amazon
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Manager / Lead
3 weeks ago
NVIDIA
Global Account Manager, Developer Relations - Google
USRemote · Full-time · Manager / Lead
3 weeks ago
NVIDIA
Senior Agentic AI For Data Science Applied Researcher
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
3 weeks ago
NVIDIA
Senior Developer Relations Manager
IN / Bengaluru, INOn-site · Full-time · Manager / Lead
3 weeks ago
Clera
ML Engineer, Agents & Reasoning
Berlin, DEOn-site · Full-time
3 weeks ago
Aging
Clera
Founding Machine Learning Engineer
Remote · Full-time
3 weeks ago
Aging
Clera
Founding Engineer – ML Research
Remote · Full-time
3 weeks ago
Aging
Clera
AI/ML Engineer
San Francisco, CA, USOn-site · Full-time
3 weeks ago
Aging
Persona AI
Machine Learning Intern, Manipulation
Houston, TX, USOn-site · Full-time · Intern
4 weeks ago
Aging
NVIDIA
Technical Marketing Engineer - AI Platform Software
Santa Clara, CA, US / US, CA, USOn-site · Full-time
4 weeks ago
NVIDIA
Engineering Manager, AI Compiler Analysis
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Manager / Lead
4 weeks ago
Simile
Research Infrastructure - Member of Technical Staff
San Francisco, CA, USOn-site · Full-time
4 weeks ago
Aging
NVIDIA
Senior Deep Learning Software Engineer, Inference and Model Optimization
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
1 month ago
Moonlake AI
Member of Technical Staff — Agent Post-Training
San Francisco, CA, USOn-site · Full-time
1 month ago
Aging
NVIDIA
Senior Performance Engineer - DGX Cloud
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
1 month ago
Lila Sciences
ML Scientist I/II, Nucleic Acid Design
San Francisco, CA, US / CA, USOn-site
$176K-$304K/yr
Aging
NVIDIA
Research Scientist, Robotics Research - PhD New College Grad 2026
Seattle, WA, US / US, WA, USOn-site · Full-time
1 month ago
Odyssey
Member of Technical Staff, Core Model Engineering
Palo Alto, CA, US / London, GBOn-site · Full-time
1 month ago
Aging