Skill
JAX AI jobs
Page 11 of 22 · 505 jax ai jobs
Open roles
NVIDIA
Senior Software Engineer, AI Networking
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
3 months ago
Aging
NVIDIA
Senior Full-Stack Lead Engineer
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
3 months ago
Aging
NVIDIA
Senior Software Engineer, CUDA Deep Learning Systems
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
3 months ago
Aging
NVIDIA
Developer Advocate – Reinforcement Learning
Santa Clara, CA, US / US, CA, USOn-site · Full-time
3 months ago
Aging
Allen Institute for AI
Research Engineer, Asta
Seattle, WA, USOn-site
3 months ago
May be filled
NVIDIA
Senior Solutions Architect, AI Hyperscalers
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
3 months ago
Aging
Cohere
Machine Learning Intern/Co-op (Fall, 2026)
CA / Europe, GBRemote · Internship · Intern
3 months ago
Aging
Rhoda AI
Inference Optimization ML Engineer
Mountain View, CA, USOn-site · Full-time
3 months ago
May be filled
Rhoda AI
Inference Infrastructure Engineer
Mountain View, CA, USOn-site · Full-time
3 months ago
May be filled
Ayo Semiconductor
Machine Learning Architect
Boston, MA, USOn-site · Full-time
3 months ago
May be filled
Ayo Semiconductor
AI Lead
Boston, MA, USOn-site · Full-time · Senior
3 months ago
May be filled
NVIDIA
Senior AI Infrastructure Software Engineer - DGX Cloud
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
3 months ago
Aging
NVIDIA
Senior Software Engineer, AI Resiliency
Redmond, WA, US / US, WA, USOn-site · Full-time · Senior
3 months ago
Aging
Cantina
Research Scientist (Singapore)
Singapore, SGOn-site · Full-time
3 months ago
May be filled
Waabi
Senior / Staff ML Training Optimization Engineer
San Francisco, CA, US / Toronto, ON, CAHybrid · Full-time · Staff / Principal
3 months ago
May be filled
NVIDIA
Senior Solutions Architect, GPU Performance and LLM - Cloud Service Providers
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
3 months ago
Aging
Harmonic
Research Engineer, Training & Inference
Palo Alto, CA, USOn-site · Full-time
$200K-$450K/yr
May be filled
Niantic Spatial
Computer Vision Researcher (VLM)
London, GBHybrid · Full-time
3 months ago
May be filled
Adaption Labs
Applied Scientist
San Francisco, CA, US / Bay AreaHybrid · Full-time
3 months ago
May be filled
SmarterDx
Staff Machine Learning Research Scientist
USRemote · Staff / Principal
3 months ago
May be filled
NVIDIA
Senior Deep Learning Tools Engineer – CUDA Tile
Santa Clara, CA, US / US, CA, USOn-site · Full-time · Senior
3 months ago
Aging
Xaira Therapeutics
AI Scientist - Biomedical Multimodal Modeling
South San Francisco, CA, USOn-site
3 months ago
Aging
Preference Model
Member of Technical Staff - Machine Learning Capabilities, New Graduates
San Francisco, CA, US / Seattle, WA, USOn-site · Full-time
$165K-$200K/yr
May be filled
Niantic Spatial
Senior Computer Vision Engineer (Localization)
San Francisco, CA, US / Sunnyvale, CA, USHybrid · Full-time · Senior
$189K-$255K/yr
May be filled