Skill
JAX AI jobs
Page 13 of 22 · 505 jax ai jobs
Open roles
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
4 months ago
Aging
xAI
Member of Technical Staff - Multimodal Understanding
Palo Alto, CA, USOn-site
4 months ago
May be filled
NVIDIA
Senior Software Engineer, JAX
CHRemote · Full-time · Senior
4 months ago
Aging
Adaption Labs
Founding Forward Deployed Machine Learning Engineer
Bay Area / San Francisco, INOn-site · Full-time
4 months ago
May be filled
Allen Institute for AI
Research Engineer, FlexOlmo
Seattle, WA, USOn-site
4 months ago
May be filled
webAI
AI Research Director
Austin, TX, USHybrid · Full-time · Director+
4 months ago
May be filled
NVIDIA
Senior Software Engineer, NCCL
Shanghai, CN / CNOn-site · Full-time · Senior
4 months ago
Aging
NVIDIA
Senior Solutions Architect, Generative AI
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
4 months ago
Aging
NVIDIA
Senior Solutions Architect, Financial Services Banking
US, NY, USOn-site · Full-time · Senior
4 months ago
Aging
webAI
Senior Forward Deployed AI Engineer
Austin, TX, US / Washington D.C. AreaOn-site · Full-time · Senior
4 months ago
May be filled
webAI
Director of Applied AI
Austin, TX, USHybrid · Full-time · Director+
4 months ago
May be filled
Goodfire
Field Team - Member of Technical Staff
San Francisco, CA, USOn-site
4 months ago
May be filled
NVIDIA
Principal Deep Learning Communication Architect
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Staff / Principal
4 months ago
Aging
NVIDIA
Senior Solutions Architect, Financial Services Capital Markets
US, NY, USRemote · Full-time · Senior
4 months ago
Aging
NVIDIA
Senior Developer Relations Manager, Robotics Physics and Simulation
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Manager / Lead
4 months ago
Aging
Manifold Bio
AI/ML Scientist, Protein Foundation Models
San Francisco, CA, US / Boston, MA, USOn-site
4 months ago
May be filled
Manifold Bio
AI/ML Scientist
San Francisco, CA, US / Boston, MA, USOn-site
4 months ago
May be filled
Manifold Bio
AI/ML Research Engineer
San Francisco, CA, US / Boston, MA, USOn-site
4 months ago
May be filled
Perplexity
Member of Technical Staff (AI Inference Engineer)
San Francisco, CA, US / Palo Alto, CA, USOn-site · Full-time
$220K-$485K/yr
Perplexity
Member of Technical Staff (AI Inference Engineer)
London, GBOn-site · Full-time
4 months ago
NVIDIA
Senior Software Engineer, Machine Learning Inference
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
4 months ago
Aging
NVIDIA
Senior Software Engineer, JAX
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
4 months ago
Aging
NVIDIA
Senior Software Engineer - Python Numerical Computing Libraries
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
4 months ago
Aging







