Skill
JAX AI jobs
Page 12 of 22 · 505 jax ai jobs
Open roles
NVIDIA
Senior Research Scientist, Multimodal Foundation Models and Robotics
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
3 months ago
Aging
Cohere
Research Internship (Fall, 2026)
San Francisco, CA, US / MontrealRemote · Full-time · Intern
3 months ago
Aging
Waabi
Distillation Lead
San Francisco, CA, US / Toronto, ON, CAHybrid · Full-time · Senior
3 months ago
May be filled
Genesis Robotics
MTS, ML Compiler and Systems
Bay AreaOn-site · Full-time
4 months ago
May be filled
Genesis Robotics
ML Compiler and Systems
Paris, FR / London, GBHybrid · Full-time
4 months ago
May be filled
NVIDIA
Tech Engagement Lead - Model Builder
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
4 months ago
Aging
Lila Sciences
ML Scientist I / II, Foundation Models for Life Sciences
San Francisco, CA, US / CA, USOn-site
$176K-$304K/yr
May be filled
Lila Sciences
Principal, Machine Learning Engineer
San Francisco, CA, US / CA, USOn-site · Staff / Principal
$252K-$374K/yr
May be filled
Lovable
Researcher, Post Training
London, GB / Stockholm, SEOn-site · Full-time
4 months ago
Aging
NVIDIA
Principal High-Performance LLM Training Engineer
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Staff / Principal
4 months ago
Aging
Lila Sciences
Senior / Principal ML Scientist, Foundation Models for Life Sciences
San Francisco, CA, US / CA, USOn-site · Staff / Principal
$268K-$384K/yr
May be filled
Arc Institute
Director, Machine Learning, Virtual Cell Initiative
Palo Alto, CA, USOn-site · Director+
4 months ago
May be filled
PhysicsX
Staff Machine Learning Software Engineer, Research
London, GBOn-site · Staff / Principal
4 months ago
May be filled
Mistral AI
Applied Scientist / Research Engineer, AI4Engineering - EMEA
Paris, FR / Munich, DEHybrid · Full-time
4 months ago
May be filled
PhysicsX
Principal Research Scientist
London, GBOn-site · Staff / Principal
4 months ago
May be filled
NVIDIA
Senior Solutions Architect, Generative AI
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
4 months ago
Aging
NVIDIA
Principal AI and ML Infra Software Engineer, GPU Clusters
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Staff / Principal
4 months ago
Aging
NVIDIA
Machine Learning Engineer - Humanoid Robotics
Shanghai, CN / CNOn-site · Full-time
4 months ago
Aging
NVIDIA
Senior Deep Learning Software Engineer
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
4 months ago
Aging
NVIDIA
Software Engineering Intern, JAX - Fall 2026
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Intern
4 months ago
Aging
Anthropic
Research Engineer, Machine Learning (RL Velocity)
London, GBOn-site
GBP 370K-GBP 630K/yr
Aging
Anthropic
Research Engineer, Machine Learning (RL Velocity)
San Francisco, CA, US / New York City, NY, USRemote
$500K-$850K/yr
Aging
RoboForce
AI Resident
Milpitas, CA, USOn-site
4 months ago
May be filled
Abridge
Clinician Scientist
SFHybrid · Full-time
$236K-$277K/yr
May be filled