JAX jobs
Open developer roles that use JAX, pulled straight from company job boards and refreshed every few hours.
405
open roles
380
posted this week
156
fully remote
144
pay $150k+
Page 4
Filter in search →
Machine Learning Research Engineer (MLRE) - Research
Achira - San Francisco, CA +1 - Hybrid
$165k–259k4 days ago
Machine Learning Research Engineer (MLRE) - Workflows/Systems
Achira - San Francisco, CA +1 - Hybrid
$165k–259k4 days ago
Machine Learning Research Engineer (MLRE) - GPUs
Achira - San Francisco, CA +1 - Hybrid
$165k–259k4 days ago
SWE - Distributed
Achira - San Francisco, CA +1 - Hybrid
$165k–259k4 days ago
Member of Technical Staff (ML)
Reka - US, UK, Remote - Remote
4 days ago
Research Engineer, Training & Inference
Harmonic - Palo Alto, CA - On-site
$200k–450k4 days ago
Research Scientist – Diffusion Models
Granica - Mountain View, CA - On-site
$160k–240k4 days ago
Research Scientist – Large Tabular Models (LTMs)
Granica - Mountain View, CA - On-site
$160k–240k4 days ago
Senior Research Scientist
adaption - San Francisco, CA - Hybrid
4 days ago
Research Scientist
adaption - New York, NY +1 - Hybrid
4 days ago
Modelling Resident
adaption - New York, NY +1 - Hybrid
4 days ago
Perception Engineer
9 Mothers - Austin, TX - On-site
$110k–300k4 days ago
ML & Molecular Simulation Scientist
Genesis Molecular AI - San Mateo, CA - Hybrid
4 days ago
AI Anime Researcher
Spellbrush - San Francisco, CA - On-site
4 days ago
Postdoctoral Fellow, MARA & Robotics
Basis Research Institute - New York, NY +1 - On-site
$100k–100k4 days ago
Research Scientist, Program Synthesis & Neuro-symbolic Methods
Basis Research Institute - New York, NY +1 - On-site
$120k–180k4 days ago
Research Scientist, World Models
Basis Research Institute - New York, NY +1 - On-site
$120k–180k4 days ago
Postdoctoral Fellow, MARA (Modeling, Abstraction and Reasoning Agents)
Basis Research Institute - Ithaca, NY +1 - On-site
$100k–100k4 days ago
Research Scientist, Machine Learning / AI
Basis Research Institute - New York, NY +1 - On-site
$120k–180k4 days ago
Research Scientist, Compilers and Programming Languages
Basis Research Institute - New York, NY +1 - On-site
$120k–180k4 days ago
Postdoctoral Fellow, Open Call
Basis Research Institute - New York, NY +1 - On-site
$100k–100k4 days ago
Staff Machine Learning Engineer, AI Infrastructure
Waymo - Mountain View, CA +2 - On-site
$251k–310k4 days ago
Controls/Robotics Intern
Terranova - Berkeley, CA - On-site
$52k–73k4 days ago
Research Scientist
Lead Allies Inc - San Francisco, CA - Hybrid
4 days ago
ML Engineer - Personalization & Recommendation Systems
Krea - San Francisco, CA - On-site
4 days ago