Skill
JAX AI jobs
Page 4 of 24 · 575 jax ai jobs
Open roles
AssemblyAI
Senior Research Engineer
NY, USRemote · Senior
1 month ago
Aging
Physical Intelligence
ML Infra Engineer, Modeling
San Francisco, CA, USOn-site · Full-time
1 month ago
Aging
NVIDIA
Senior Performance Engineer - DGX Cloud
IN / Bengaluru, INOn-site · Full-time · Senior
1 month ago
NVIDIA
Senior HPC and AI Network Software Architect
Zurich / CHOn-site · Full-time · Senior
1 month ago
NVIDIA
Senior System Software Architect, HPC and AI Networking
Shanghai, CN / CNOn-site · Full-time · Senior
1 month ago
Clera
Founding Machine Learning Engineer
Mountain View, CA, USOn-site · Full-time
1 month ago
Aging
AfterQuery
Research Engineer
San Francisco, CA, USOn-site · Full-time
$210K-$450K/yr
Aging
NVIDIA
Senior Solutions Architect, Robotics Foundation Model Training
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
1 month ago
AMI
AMI Scientist/Interns - Geometry and 3D Vision
Montreal / Paris, NY, SGOn-site · Full-time · Intern
1 month ago
Aging
Institute of Foundation Models
Community Development Manager
Sunnyvale, CA, USOn-site · Full-time · Manager / Lead
$150K-$350K/yr
Aging
Robotics and AI Institute
Internship - Software Engineering
ZurichOn-site · Other · Intern
1 month ago
Aging
Clera
Founding Engineer - ML Research
Mountain View, CA, USOn-site · Full-time
1 month ago
Aging
Tessera Labs
Research Engineer
New York City, NY, US / San Jose Office (HQ)On-site · Full-time
$200K-$300K/yr
Aging
Clera
Founding Machine Learning Engineer
Mountain View, CA, USOn-site · Full-time
1 month ago
Aging
Neurophos
Senior ML Engineer, Optimization
Austin, TX, USOn-site · Full-time · Senior
$160K-$215K/yr
Aging
OpenAI
Software Engineer, Trainium
San Francisco, CA, USHybrid · Full-time
$295K-$380K/yr
May be filled
NVIDIA
Deep Learning Performance Architect
Shanghai, CN / CNOn-site · Full-time
1 month ago
NVIDIA
NVIDIA 2027 Internships: Ph.D. Research Robotics
Santa Clara, CA, US / US, CA, USOn-site · Full-time
1 month ago
NVIDIA
NVIDIA 2027 Internships: Ph.D. Research Large Language Models
Santa Clara, CA, US / US, CA, USOn-site · Full-time
1 month ago
NVIDIA
NVIDIA 2027 Internships: Ph.D. Research Graphics and Simulation
Santa Clara, CA, US / US, CA, USOn-site · Full-time
1 month ago
NVIDIA
NVIDIA 2027 Internships: Ph.D. Research Generative AI
Santa Clara, CA, US / US, CA, USOn-site · Full-time
1 month ago
Baseten
Forward Deployed Engineer (Training)
San Francisco, CA, USHybrid · Full-time
$200K-$400K/yr
Aging
Genesis Robotics
Genesis-World: Core Simulation Engine Engineer
Bay AreaOn-site · Full-time
1 month ago
Aging
Nimble Robotics
Senior/Staff Software Engineer, Infrastructure (ML)
San Francisco, CA, USOn-site · Staff / Principal
1 month ago
Aging