JAX jobs in California, United States

Discover JAX roles in California, United States on Get AI Careers and apply today.

14 jobs
Gauss Labs logoGL

AI Scientist - Machine Learning (US/KR)

Gauss Labs

Palo Alto, California, United States (Hybrid)

2 wk. ago
AC

Machine Learning Research Engineer (MLRE) - Workflows/Systems

Achira

San Francisco, California, United States (Hybrid)

5 days ago
Abridge logoAB

Machine Learning Scientist (All Levels)

Abridge

San Francisco, California, United States (Hybrid)

$205–300K Yearly5 days ago
Zoox logoZO

Senior Engineering Manager, ML Platform

Zoox

Foster City, California, United States (Hybrid)

$317–370K Yearly2 wk. ago
Centific logoCE

Research Scientist, LLM Evaluation & Post-Training

Centific

Palo Alto, California, United States (Hybrid)

$150–300K Yearly2 wk. ago
Plaud logoPL

Machine Learning Engineer, Speech LLM Training - San Francisco

Plaud

San Francisco, California, United States (Hybrid)

5 days ago
AC

Machine Learning Research Engineer (MLRE) - Research

Achira

San Francisco, California, United States (Hybrid)

5 days ago
Zoox logoZO

Machine Learning Engineer - Semantic Reasoning (Highway)

Zoox

Foster City, California, United States (Hybrid)

$189–258K Yearly4 days ago
Zillow Group logoZG

Machine Learning Engineer

Zillow Group

United States (Remote)

4 wk. ago
Hugging Face logoHF

Open-Source Machine Learning Engineer - US Remote

Hugging Face

United States (Remote)

5 days ago
Character.AI logoCH

Machine Learning Infrastructure Engineer

Character.AI

United States (Remote)

$150–350K Yearly5 days ago
SandboxAQ logoSA

Senior Machine Learning Engineer, AI Generation Engine

SandboxAQ

United States (Remote)

$164–246K Yearly5 days ago
SandboxAQ logoSA

Staff Machine Learning Engineer, AI Generation Engine

SandboxAQ

United States (Remote)

$196–294K Yearly5 days ago