Skill
JAX AI jobs
Page 8 of 22 · 505 jax ai jobs
Open roles
NVIDIA
Senior Deep Learning Framework Communications Engineer
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
2 months ago
Aging
Zoox
Machine Learning Engineer - Simulation Framework
Seattle, WA, US / Foster City, CA, USHybrid · Full-time
$151K-$257K/yr
May be filled
Zoox
Senior/Staff Software Engineer - C++ Simulation Platform
Seattle, WA, US / Foster City, CA, USHybrid · Full-time · Staff / Principal
$245K-$305K/yr
May be filled
xAI
Member of Technical Staff
Palo Alto, CA, USOn-site
2 months ago
May be filled
NVIDIA
Software Engineer, AI and DL Kernel Libraries
Shanghai, CN / CNOn-site · Full-time
2 months ago
Aging
NVIDIA
Senior Solutions Architect, Generative AI Research
US, FL, USRemote · Full-time · Senior
2 months ago
Aging
NVIDIA
Senior Solutions Architect, AI - Accelerated Physics
US, CA, USRemote · Full-time · Senior
2 months ago
Aging
NVIDIA
Senior Software Engineer, AI Storage
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
2 months ago
Aging
NVIDIA
Senior Systems Software Engineer, AI Stack and Performance - DGX Station
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
2 months ago
Aging
NVIDIA
Senior Manager, Software Development - GPU Accelerated Storage
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Manager / Lead
2 months ago
Aging
Innodata
Applied Research Scientist, LLM Evaluation & Post-Training
USRemote
2 months ago
May be filled
Innodata
AI/ML Research Engineer, LLM Post-Training & Evaluation
USRemote
2 months ago
May be filled
Persona AI
Reinforcement Learning Engineer, Grasping
Houston, TX, USOn-site · Full-time
2 months ago
May be filled
1X
Software Engineer - Simulation
San Carlos, CA, USOn-site · Full-time
$200K-$280K/yr
May be filled
1X
AI Residency
San Carlos, CA, USOn-site · Temporary
$10K+/month
May be filled
Writer
Staff AI research scientist
San Francisco, CA, US / Seattle, WA, USHybrid · Full-time · Staff / Principal
$234.3K-$296K/yr
Aging
Diffractive Labs
Research Scientist/Engineer, GNNs
London, GBHybrid · Full-time
2 months ago
Diffractive Labs
Research Engineer, Machine Learning
London, GBHybrid · Full-time
2 months ago
Snowflake
AI System Research and Development Engineer - Optimization
US-WA-BellevueOn-site · Full-time
$200K-$265K/yr
May be filled
Snowflake
AI Research Scientist, New Grad – Agents & Reinforcement Learning
US-WA-BellevueOn-site · Full-time · Entry / Junior
$176K-$230K/yr
May be filled
Periodic Labs
Forward Deployed Engineer, Quantum Simulations
Menlo Park, CA, USOn-site · Full-time
2 months ago
May be filled
NVIDIA
Senior Deep Learning Frameworks CUDA Software Engineer
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
2 months ago
Aging
Together AI
Research Intern, Model Shaping (Fall 2026)
San Francisco, CA, USOn-site · Internship · Intern
2 months ago
Aging
NVIDIA
High-Performance LLM Training Engineer - New College Grad 2026
Santa Clara, CA, US / US, CA, USOn-site · Full-time
2 months ago
Aging