Skill
JAX AI jobs
Page 10 of 22 · 505 jax ai jobs
Open roles
Clera
AI/ML Engineer
San Francisco, CA, USOn-site · Full-time
2 months ago
Runway
Member of Technical Staff, Robotics Research Engineer
NY, USOn-site · Full-time
$270K-$370K/yr
May be filled
NVIDIA
Senior AI Software Engineer, Kernel Libraries
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
2 months ago
Aging
NVIDIA
Senior Software Engineer, AI and DL Kernel Libraries
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
2 months ago
Aging
NVIDIA
Senior Software Engineer, CUDA Core Libraries
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
2 months ago
Aging
NVIDIA
Senior Software Engineer, CUTLASS Performance
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
2 months ago
Aging
Adaption Labs
Senior Research Scientist
San Francisco, CA, US / United States, NY, USRemote · Full-time · Senior
2 months ago
Zoox
Machine Learning Engineer - Semantic Reasoning (Highway)
Foster City, CA, US / Boston, MA, USHybrid · Full-time
$189K-$258K/yr
May be filled
Lila Sciences
Principal Software Engineer, Data
San Francisco, CA, US / Cambridge, MA, USOn-site · Staff / Principal
$204K-$348K/yr
May be filled
NVIDIA
Developer Advocate – Distributed Training
Santa Clara, CA, US / US, CA, USOn-site · Full-time
3 months ago
Lila Sciences
Senior Software Engineer, App
San Francisco, CA, US / Cambridge, MA, USOn-site · Senior
$144K-$240K/yr
May be filled
Lila Sciences
Senior Software Engineer, Data
San Francisco, CA, US / Cambridge, MA, USOn-site · Senior
$144K-$288K/yr
May be filled
NVIDIA
Software Engineering Intern, Robot Learning Platform - 2026
Shanghai, CN / CNOn-site · Full-time · Intern
3 months ago
Aging
NVIDIA
Senior Software Engineer — cuEquivariance
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
3 months ago
Aging
Lila Sciences
Sr Principal/Principal Software Engineer, App
San Francisco, CA, US / Cambridge, MA, USOn-site · Staff / Principal
$204K-$348K/yr
May be filled
NVIDIA
Senior DGX Cloud AI Infrastructure Software Engineer
Shanghai, CN / CNOn-site · Full-time · Senior
3 months ago
Aging
NVIDIA
Senior Quantum AI Research Scientist, Applied Research
Redmond, WA, US / US, WA, USOn-site · Full-time · Senior
3 months ago
Aging
NVIDIA
Developer Relations Manager, Higher Education and Research - Foundational AI
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Manager / Lead
3 months ago
Aging
Rhoda AI
Research Member of Technical Staff- Video Generation Modeling
Mountain View, CA, USOn-site · Full-time
3 months ago
May be filled
Rhoda AI
Research Member of Technical Staff- Reasoning
Mountain View, CA, USOn-site · Full-time
3 months ago
May be filled
Writer
AI engineer (UK)
London, GBHybrid · Full-time
3 months ago
Aging
Writer
AI engineer
San Francisco, CA, US / Seattle, WA, USHybrid · Full-time
$167.9K-$328.5K/yr
Aging
Rhoda AI
Research Member of Technical Staff- Training Systems
Mountain View, CA, USOn-site · Full-time
3 months ago
May be filled
Meshy
Generative AI - ML System Engineering
Shanghai, CNHybrid · Full-time
3 months ago
May be filled