Skill
JAX AI jobs
Page 4 of 22 · 505 jax ai jobs
Open roles
NVIDIA
System Software Engineer, Profiler - GPU
Austin, TX, US / US, TX, USOn-site · Full-time
1 month ago
Innodata
Applied Research Scientist, LLM Evaluation & Post-Training
CARemote
1 month ago
Aging
Innodata
AI/ML Research Engineer, LLM Post-Training & Evaluation
CARemote
1 month ago
Aging
NVIDIA
Senior Software Engineer, CUDA Core Libraries
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
1 month ago
NVIDIA
AI and ML Infra Software Engineer, GPU Clusters - New College Grad 2026
Santa Clara, CA, US / US, CA, USOn-site · Full-time
1 month ago
Cerebras
Director/Sr. Manager, AI Inference Model Scaling
US and Canada OfficesOn-site · Full-time · Director+
1 month ago
Cerebras
ML Software Engineer - Integration & Quality - New Grad
US and Canada OfficesHybrid · Internship · Entry / Junior
1 month ago
NVIDIA
Senior Technical Program Manager, Deep Learning Frameworks
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
1 month ago
xAI
ML Infrastructure Engineer
Palo Alto, CA, USOn-site
1 month ago
Aging
NVIDIA
Senior HPC Cluster Administrator - Deep Learning Frameworks Infrastructure
Warsaw, PL / PLOn-site · Full-time · Senior
1 month ago
Brain Co.
AI/ML Engineer, Deployed
London, GBHybrid · Full-time
1 month ago
Aging
Radical Numerics
Member of Technical Staff, Structure Prediction
San Francisco, CA, USOn-site · Full-time
1 month ago
Aging
Cohere
Engineering Manager, GPU Infrastructure
United States, CAOn-site · Full-time · Manager / Lead
1 month ago
Aging
Dyna Robotics
Applied Researcher - Deployment Intelligence & Continuous Learning
Redwood City, CA, USOn-site · Full-time
1 month ago
Aging
Zoox
Software Engineer, ML Platform (ML Training)
Foster City, CA, USHybrid · Full-time
$182K-$257K/yr
Aging
NVIDIA
Deep Learning Software Engineer, TensorRT Performance - New College Grad 2026
Santa Clara, CA, US / US, CA, USOn-site · Full-time
1 month ago
Causal Labs
Member of Technical Staff — Inference Infrastructure
San Francisco, CA, USOn-site · Full-time
1 month ago
Aging
Clera
Research Scientist / Research Engineer
San Francisco, CA, USOn-site · Full-time
1 month ago
Aging
Lila Sciences
Senior ML Scientist, Biological Systems
San Francisco, CA, US / CA, USOn-site · Senior
$268K-$336K/yr
Aging
NVIDIA
Robotics Software Engineer - Robot Simulation and Benchmarking
Zurich / CHOn-site · Full-time
1 month ago
NVIDIA
Deep Learning Performance Architect
Shanghai, CN / CNOn-site · Full-time
1 month ago
P-1 AI
AI Research Scientist/Manager - Adaptive Intelligence
San Francisco Bay AreaHybrid · Full-time · Manager / Lead
$200K-$385K/yr
Aging
P-1 AI
AI Research Scientist/Manager - Spatial Reasoning
San Francisco Bay AreaHybrid · Full-time · Manager / Lead
$200K-$385K/yr
Aging
Armada
Technical Trainer - AI Infrastructure Products (Part-time)
USRemote
1 month ago
Aging