Skill
JAX AI jobs
Page 17 of 24 · 575 jax ai jobs
Open roles
NVIDIA
Senior High-Performance LLM Training Engineer
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
6 months ago
Aging
NVIDIA
Senior Robotics Research Engineer, Robotics and AI for Drug Discovery
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
6 months ago
Aging
Crusoe
Staff Software Engineer, Storage
San Francisco, CA, US / CA, USOn-site · Full-time · Staff / Principal
6 months ago
Crusoe
Senior Staff Software Engineer, Storage
San Francisco, CA, US / CA, USOn-site · Full-time · Staff / Principal
6 months ago
Perplexity
Internship - Search Machine Learning Engineer
London, GBOn-site · Full-time · Intern
6 months ago
Aging
NVIDIA
Solutions Architect, LLM Model Builder
Santa Clara, CA, US / US, CA, USOn-site · Full-time
6 months ago
Aging
NVIDIA
Solutions Architect, LLM Model Builder
Santa Clara, CA, US / US, CA, USOn-site · Full-time
6 months ago
Aging
NVIDIA
Senior Manager, Software Engineering - JAX
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Manager / Lead
6 months ago
Aging
Waabi
Research Engineer, World Models
San Francisco, CA, US / Toronto, ON, CAHybrid · Full-time
6 months ago
May be filled
NVIDIA
Senior DGX Cloud AI Infrastructure Software Engineer
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
6 months ago
Aging
Xaira Therapeutics
BioMedical AI Research Engineer
South San Francisco, CA, USOn-site
6 months ago
May be filled
NVIDIA
Deep Learning Algorithms Engineer - ACOT
Ho Chi Minh City, VN / VNOn-site · Full-time
6 months ago
Aging
Armada
Senior Customer Solutions Engineer, AI Factory, ANZ
Sydney, AUOn-site · Senior
6 months ago
May be filled
Waabi
Research Scientist, World Models
San Francisco, CA, US / USHybrid · Full-time
6 months ago
May be filled
Waabi
Research Scientist, Neural Reconstruction
San Francisco, CA, US / USHybrid · Full-time
6 months ago
May be filled
Waabi
Senior / Staff Software Engineer, ML Datasets & Data Pipelines
San Francisco, CA, US / Toronto, ON, CAHybrid · Full-time · Staff / Principal
6 months ago
May be filled
Waabi
Research Scientist, Simulation Agents
San Francisco, CA, US / Toronto, ON, CAHybrid · Full-time
6 months ago
May be filled
DeepMind
Research Engineer, Human Understanding
Mountain View, CA, US / Los Angeles, CA, USOn-site
6 months ago
Aging
Eragon
Machine Learning Engineer
San Francisco, CA, USOn-site · Full-time
$180K-$230K/yr
May be filled
Lila Sciences
Sr Principal/ Principal Software Engineer, AI Lab Execution System
San Francisco, CA, US / Cambridge, MA, USOn-site · Staff / Principal
$204K-$348K/yr
May be filled
Anthropic
Research Engineer, Performance RL (Reinforcement Learning)
San Francisco, CA, USOn-site
$350K-$850K/yr
May be filled
Baseten
Post-Training Research Engineer
San Francisco, CA, USHybrid · Full-time
$200K-$275K/yr
May be filled
Cohere
Senior Member of Technical Staff, Safety for Agents
Paris, FR / MontrealHybrid · Full-time · Senior
6 months ago
May be filled
NVIDIA
Senior Deep Learning Software Engineer, TensorRT Performance
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
6 months ago
Aging