Skill
JAX AI jobs
Page 16 of 22 · 505 jax ai jobs
Open roles
Mistral AI
Research Engineer, Machine Learning
Palo Alto, CA, USHybrid · Full-time
7 months ago
May be filled
Physical Intelligence
ML Infra Engineer (TPU/Jax/Optimization)
San Francisco, CA, USOn-site · Full-time
7 months ago
May be filled
Featherless AI
AI Researcher – Multilingual Data
worldRemote · Full-time
7 months ago
May be filled
Featherless AI
AI Researcher — AI Architecture Research
worldRemote · Full-time
7 months ago
May be filled
Udio
Member of Technical Staff - Quantitative Research
New York City ( possible for exceptional candidatesRemote
7 months ago
May be filled
Featherless AI
Machine Learning Engineer — Distillation
worldRemote · Full-time
7 months ago
May be filled
Featherless AI
Machine Learning Engineer — AI Architecture Research
worldRemote · Full-time
7 months ago
May be filled
Brain Co.
Early Career AI/ML Engineer
San Francisco Bay AreaHybrid · Full-time
7 months ago
May be filled
Dexmate
Reinforcement learning engineer
Fremont, CA, USOn-site · Full-time
$120K-$300K/yr
May be filled
Anthropic
Research Engineer / Research Scientist, Vision
San Francisco, CA, US / Seattle, WA, USOn-site
$350K-$850K/yr
Aging
Anthropic
Research Engineer/Research Scientist, Audio
San Francisco, CA, USOn-site
$350K-$500K/yr
Aging
Goodfire
Head of Engineering, Platform
San Francisco, CA, USOn-site · Director+
7 months ago
May be filled
Cohere
Software Engineer, GPU Infrastructure (HPC)
Canada, USHybrid · Full-time
7 months ago
Aging
Perplexity
Internship - Search Machine Learning Engineer
Belgrade, RSOn-site · Full-time · Intern
7 months ago
OpenAI
ML Research Engineer - Hardware Codesign
San Francisco, CA, USOn-site · Full-time
$185K-$455K/yr
May be filled
Runway
Member of Technical Staff, Research Engineer
Remote · Full-time
$270K-$370K/yr
May be filled
webAI
AI Research Scientist
Austin, TX, USHybrid · Full-time
8 months ago
May be filled
Zoox
Senior Machine Learning Engineer - Simulation Scenario Generation
Foster City, CA, USHybrid · Full-time · Senior
$233K-$290K/yr
May be filled
Hammerhead AI
Reinforcement Learning Engineer
Redwood City, CA, USOn-site · Full-time
8 months ago
May be filled
Krea
ML Engineer - Personalization & Recommendation Systems
San Francisco, CA, USOn-site · Full-time
8 months ago
May be filled
Graphon.AI
ML Infrastructure Engineer
San Francisco, CA, USOn-site
8 months ago
May be filled
Dexmate
Robot learning engineer
Fremont, CA, USOn-site · Full-time
$100K-$300K/yr
May be filled
Cohere
Member of Technical Staff, Data Analysis and Evaluation
Paris, FR / MontrealRemote · Full-time
8 months ago
Aging
Skild AI
Computer Vision AI & ML Engineer
Bengaluru, INOn-site
8 months ago
May be filled