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 8
Filter in search →
Applied AI Research Engineer
Netic - San Francisco, CA - On-site
$145k–330k4 days ago
Sr. Staff ML Researcher, LLM Algorithmic Optimization
d-Matrix - Santa Clara, CA - On-site
$175k–265k4 days ago
Compiler Frontend Engineer
Normal Computing Corporation - Mountain View, CA +1 - Hybrid
4 days ago
Member of Product Staff, Life Sciences
Abridge - San Francisco, CA +2 - Hybrid
$240k–260k4 days ago
Engineering Leadership, NY Site Lead
Abridge - New York, NY - On-site
$290k–350k4 days ago
Member of Product Staff, AI/ML (Evals)
Abridge - San Francisco, CA - Hybrid
$250k–290k4 days ago
Member of Product Operations & Strategy Staff
Abridge - San Francisco, CA - Hybrid
$150k–165k4 days ago
Member of Technical Staff, Research Science
Abridge - New York, NY - Hybrid
$188k–277k4 days ago
Member of Design Staff, Product Design
Abridge - San Francisco, CA +1 - Hybrid
$218k–299k4 days ago
Member of Technical Staff, Machine Learning
Abridge - San Francisco, CA +1 - Hybrid
$205k–300k4 days ago
Data Scientist
Fuse - San Leandro, CA - On-site
$145k–190k4 days ago
Technical Support Engineer
TensorWave - Las Vegas, NV - On-site
4 days ago
Member of Technical Staff - Pre-Training Data
Reflection - San Francisco, CA - On-site
4 days ago
Senior Inference Optimization ML Engineer
Rhoda AI - Mountain View, CA - On-site
4 days ago
Research Member of Technical Staff- Video Generation Modeling
Rhoda AI - Mountain View, CA - On-site
4 days ago
Software Engineer, Inference Infrastructure
Rhoda AI - Mountain View, CA - On-site
$175k–280k4 days ago
Research Member of Technical Staff- Training Systems
Rhoda AI - Mountain View, CA - On-site
$200k–300k4 days ago
Senior Roboticist
Astera Institute - Emeryville, CA - On-site
$250k–500k4 days ago
Optical Engineer, Precision Instrumentation
Astera Institute - Alameda, CA - On-site
$150k–250k4 days ago
Research Scientist, Theory & Computation
Astera Institute - Alameda, CA - On-site
$150k–300k4 days ago
Senior/Staff AI Scientist (Polytope Bio)
Astera Institute - Emeryville, CA - Hybrid
$200k–300k4 days ago
Precision Instrumentation Control & Estimation Consultant
Astera Institute - Emeryville, CA - Hybrid
4 days ago
Principal AI Scientist (Polytope Bio)
Astera Institute - Remote - Remote
$250k–350k4 days ago
Research Scientist: Hierarchical Sensorimotor Perception
Astera Institute - Emeryville, CA - On-site
$150k–300k4 days ago
Research Scientist: Energy Based Models
Astera Institute - Emeryville, CA - On-site
$150k–300k4 days ago