Skill
JAX AI jobs
Page 2 of 24 · 572 jax ai jobs
Open roles
Chai Discovery
Research Engineer - Auto Research
San Francisco, CA, USOn-site · Full-time
2 weeks ago
Aging
Chai Discovery
Research Engineer - ML Infrastructure
San Francisco, CA, USOn-site · Full-time
2 weeks ago
Aging
NVIDIA
PhD Research Intern, Quantum Simulation and AI - 2027
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Intern
2 weeks ago
NVIDIA
Senior Deep Learning Compiler Engineer - PyTorch
Berlin, DE / DEOn-site · Full-time · Senior
2 weeks ago
Writer
AI research scientist
San Francisco, CA, US / Seattle, WA, USHybrid · Full-time
$199K-$325K/yr
Writer
AI engineer (UK)
London, GBHybrid · Full-time
3 weeks ago
Writer
AI engineer
San Francisco, CA, US / New York City, NY, USHybrid · Full-time
$198K-$259K/yr
NVIDIA
Research Intern, Robotics - Summer 2027
Seattle, WA, US / US, WA, USOn-site · Full-time · Intern
3 weeks ago
NVIDIA
Developer Relations Manager, Higher Education and Research - Foundational AI
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Manager / Lead
3 weeks ago
NVIDIA
Software Engineering Intern, DLFW Comms - 2027
Shanghai, CN / CNOn-site · Full-time · Intern
3 weeks ago
Together AI
Research Intern, Frontier Agents (Summer 2027)
San Francisco, CA, USOn-site · Internship · Intern
3 weeks ago
Aging
Together AI
Research Intern, Frontier Agents (Winter 2027)
San Francisco, CA, USOn-site · Internship · Intern
3 weeks ago
Aging
Together AI
Research Intern, Model Shaping (Winter 2027)
San Francisco, CA, USOn-site · Internship · Intern
3 weeks ago
Aging
Together AI
Research Intern, Model Shaping (Summer 2027)
San Francisco, CA, USOn-site · Internship · Intern
3 weeks ago
Aging
Together AI
Research Intern, Inference (Winter 2027)
San Francisco, CA, USOn-site · Internship · Intern
3 weeks ago
Aging
Together AI
Research Intern, Inference (Summer 2027)
San Francisco, CA, USOn-site · Internship · Intern
3 weeks ago
Aging
NVIDIA
Research Scientist, Networking Research - PhD New College Grad 2026
Santa Clara, CA, US / US, CA, USOn-site · Full-time
3 weeks ago
NVIDIA
PhD Research Intern, Generalist Embodied Agents Research - 2027
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Intern
3 weeks ago
NVIDIA
Senior Deep Learning Compiler Engineer - XLA
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
3 weeks ago
NVIDIA
Senior Deep Learning Engineer, Cosmo 3D Spatial
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
4 weeks ago
Absentia Labs
AI Research Scientist
Boston, MA, USHybrid · Contract
$130K-$208K/yr
Aging
NVIDIA
Software Engineer, Robotics - Isaac Lab
Shanghai, CN / CNOn-site · Full-time
4 weeks ago
NVIDIA
Humanoid Robot Learning Intern, Isaac Loco-Manipulation - 2027
Shanghai, CN / CNOn-site · Full-time · Intern
4 weeks ago
Preference Model
Member of Technical Staff - Research & Post-training
San Francisco, CA, USOn-site · Full-time
$200K-$350K/yr
Aging