Skill
JAX AI jobs
Page 15 of 24 · 570 jax ai jobs
Open roles
NVIDIA
Tech Engagement Lead - Model Builder
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
5 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
5 months ago
Aging
NVIDIA
Principal High-Performance LLM Training Engineer
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Staff / Principal
5 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+
5 months ago
May be filled
PhysicsX
Staff Machine Learning Software Engineer, Research
London, GBOn-site · Staff / Principal
5 months ago
May be filled
Mistral AI
Applied Scientist / Research Engineer, AI4Engineering - EMEA
Paris, FR / Munich, DEHybrid · Full-time
5 months ago
Aging
PhysicsX
Principal Research Scientist
London, GBOn-site · Staff / Principal
5 months ago
May be filled
NVIDIA
Senior Solutions Architect, Generative AI
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
5 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
5 months ago
Aging
NVIDIA
Machine Learning Engineer - Humanoid Robotics
Shanghai, CN / CNOn-site · Full-time
5 months ago
Aging
NVIDIA
Senior Deep Learning Software Engineer
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
5 months ago
Aging
NVIDIA
Software Engineering Intern, JAX - Fall 2026
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Intern
5 months ago
Aging
Anthropic
Research Engineer, Machine Learning (RL Velocity)
London, GBOn-site
GBP 370K-GBP 630K/yr
May be filled
Anthropic
Research Engineer, Machine Learning (RL Velocity)
San Francisco, CA, US / New York City, NY, USRemote
$500K-$850K/yr
May be filled
RoboForce
AI Resident
Milpitas, CA, USOn-site
5 months ago
May be filled
Abridge
Clinician Scientist
SFHybrid · Full-time
$236K-$277K/yr
May be filled
Forus
Research Engineer
NY, USOn-site · Full-time
$170K-$300K/yr
May be filled
NVIDIA
Solutions Architect, AI Models
Santa Clara, CA, US / US, CA, USOn-site · Full-time
5 months ago
Aging
xAI
Member of Technical Staff - Multimodal Understanding
Palo Alto, CA, USOn-site
5 months ago
Aging
NVIDIA
Senior Software Engineer, JAX
CHRemote · Full-time · Senior
5 months ago
Aging
Adaption Labs
Founding Forward Deployed Machine Learning Engineer
Bay Area / San Francisco, INOn-site · Full-time
6 months ago
May be filled