Skill
JAX AI jobs
Page 14 of 22 · 505 jax ai jobs
Open roles
NVIDIA
Senior High-Performance LLM Training Engineer
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
4 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
4 months ago
Aging
Crusoe
Staff Software Engineer, Storage
San Francisco, CA, US / CA, USOn-site · Full-time · Staff / Principal
4 months ago
May be filled
Crusoe
Senior Staff Software Engineer, Storage
San Francisco, CA, US / CA, USOn-site · Full-time · Staff / Principal
4 months ago
May be filled
Perplexity
Internship - Search Machine Learning Engineer
London, GBOn-site · Full-time · Intern
4 months ago
NVIDIA
Solutions Architect, LLM Model Builder
Santa Clara, CA, US / US, CA, USOn-site · Full-time
4 months ago
Aging
NVIDIA
Solutions Architect, LLM Model Builder
Santa Clara, CA, US / US, CA, USOn-site · Full-time
4 months ago
Aging
NVIDIA
Senior Manager, Software Engineering - JAX
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Manager / Lead
4 months ago
Aging
Waabi
Research Engineer, World Models
San Francisco, CA, US / Toronto, ON, CAHybrid · Full-time
4 months ago
May be filled
NVIDIA
Senior DGX Cloud AI Infrastructure Software Engineer
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
4 months ago
Aging
Xaira Therapeutics
BioMedical AI Research Engineer
South San Francisco, CA, USOn-site
4 months ago
Aging
NVIDIA
Deep Learning Algorithms Engineer - ACOT
Ho Chi Minh City, VN / VNOn-site · Full-time
5 months ago
Aging
Armada
Senior Customer Solutions Engineer, AI Factory, ANZ
Sydney, AUOn-site · Senior
5 months ago
May be filled
Waabi
Research Scientist, World Models
San Francisco, CA, US / USHybrid · Full-time
5 months ago
May be filled
Waabi
Research Scientist, Neural Reconstruction
San Francisco, CA, US / USHybrid · Full-time
5 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
5 months ago
May be filled
Waabi
Research Scientist, Simulation Agents
San Francisco, CA, US / Toronto, ON, CAHybrid · Full-time
5 months ago
May be filled
DeepMind
Research Engineer, Human Understanding
Mountain View, CA, US / Los Angeles, CA, USOn-site
5 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
Aging
Baseten
Post-Training Research Engineer
San Francisco, CA, USHybrid · Full-time
$200K-$275K/yr
Cohere
Senior Member of Technical Staff, Safety for Agents
Paris, FR / MontrealHybrid · Full-time · Senior
5 months ago
Aging
NVIDIA
Senior Deep Learning Software Engineer, TensorRT Performance
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
5 months ago
Aging








