Skill
JAX AI jobs
Page 19 of 24 · 575 jax ai jobs
Open roles
Mistral AI
Research Engineer, Machine Learning
Palo Alto, CA, USHybrid · Full-time
8 months ago
Aging
Physical Intelligence
ML Infra Engineer (TPU/Jax/Optimization)
San Francisco, CA, USOn-site · Full-time
8 months ago
May be filled
Featherless AI
AI Researcher – Multilingual Data
worldRemote · Full-time
8 months ago
May be filled
Featherless AI
AI Researcher — AI Architecture Research
worldRemote · Full-time
8 months ago
Aging
Udio
Member of Technical Staff - Quantitative Research
New York City ( possible for exceptional candidatesRemote
8 months ago
May be filled
Featherless AI
Machine Learning Engineer — Distillation
worldRemote · Full-time
8 months ago
Aging
Featherless AI
Machine Learning Engineer — AI Architecture Research
worldRemote · Full-time
8 months ago
Aging
Brain Co.
Early Career AI/ML Engineer
San Francisco Bay AreaHybrid · Full-time
8 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
May be filled
Anthropic
Research Engineer/Research Scientist, Audio
San Francisco, CA, USOn-site
$350K-$500K/yr
May be filled
Goodfire
Head of Engineering, Platform
San Francisco, CA, USOn-site · Director+
9 months ago
May be filled
Cohere
Software Engineer, GPU Infrastructure (HPC)
Canada, USHybrid · Full-time
9 months ago
May be filled
Perplexity
Internship - Search Machine Learning Engineer
Belgrade, RSOn-site · Full-time · Intern
9 months ago
Aging
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
9 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
9 months ago
May be filled
Krea
ML Engineer - Personalization & Recommendation Systems
San Francisco, CA, USOn-site · Full-time
10 months ago
May be filled
Graphon.AI
ML Infrastructure Engineer
San Francisco, CA, USOn-site
10 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
10 months ago
May be filled
Skild AI
Computer Vision AI & ML Engineer
Bengaluru, INOn-site
10 months ago
May be filled