Skill
JAX AI jobs
Page 12 of 24 · 570 jax ai jobs
Open roles
Adaption Labs
Modelling Resident (8-Month Contract)
San Francisco, CA, US / Global, NY, USRemote · Contract
4 months ago
May be filled
Adaption Labs
Applied ML Engineer
San Francisco, SGOn-site · Full-time
4 months ago
May be filled
Lila Sciences
Co-op, Machine Learning for Digital Twins
Cambridge, MA, US / MA, USOn-site
4 months ago
May be filled
Xaira Therapeutics
Software Engineer, ML Platform
Seattle, WA, USOn-site
4 months ago
May be filled
Snowflake
Post-Doctoral Researcher (Fixed-Term)
US-WA-BellevueOn-site · Full-time
$160K-$220K/yr
May be filled
NVIDIA
Solutions Architect, Financial Services Capital Markets
US, NY, USRemote · Full-time
4 months ago
Aging
NVIDIA
Deep Learning Performance Architect
Shanghai, CN / CNOn-site · Full-time
4 months ago
Aging
NVIDIA
Senior Machine Learning Engineer, End‑to‑End Autonomous Driving
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
4 months ago
Aging
Ellipsis Health
Senior ML Engineer, Agentic AI
San Francisco (hybrid)Hybrid · Full-time · Senior
$160K-$210K/yr
May be filled
Allen Institute for AI
Senior Research Engineer, Olmo + Molmo
Seattle, WA, USOn-site · Senior
4 months ago
May be filled
NVIDIA
Senior DL Software Engineer, Model Optimization and Edge Deployment - Autonomous Vehicles
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
4 months ago
Aging
NVIDIA
Senior Software Engineer - Developer Tools for Deep Learning
US, MA, USRemote · Full-time · Senior
4 months ago
Aging
NVIDIA
Machine Learning Intern, Humanoid Robotics - 2026
Shanghai, CN / CNOn-site · Full-time · Intern
4 months ago
Aging
KoBold Metals
Postdoctoral Scholar — AI Researcher for Critical Mineral Discovery
Remote
4 months ago
May be filled
Skild AI
Computer Vision AI & ML Engineer
San Mateo, CA, USOn-site
4 months ago
May be filled
Mach9
ML Engineer
San Francisco, CA, USOn-site · Full-time
4 months ago
May be filled
NVIDIA
Software Engineering Manager, Robotics Neural Reconstruction and Real2Sim Applications
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Manager / Lead
4 months ago
Aging
NVIDIA
AI Inference Performance Engineer - New College Grad 2026
Santa Clara, CA, US / US, CA, USOn-site · Full-time
4 months ago
Aging
NVIDIA
Senior Deep Learning Performance Architect
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
4 months ago
Aging
Clera
Senior AI/ML Engineer
San Francisco, CA, USOn-site · Full-time · Senior
4 months ago
May be filled
Clera
AI/ML Engineer
San Francisco, CA, USOn-site · Full-time
4 months ago
May be filled
Runway
Member of Technical Staff, Robotics Research Engineer
NY, USOn-site · Full-time
$270K-$370K/yr
May be filled
NVIDIA
Senior AI Software Engineer, Kernel Libraries
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
4 months ago
Aging
NVIDIA
Senior Software Engineer, AI and DL Kernel Libraries
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
4 months ago
Aging