JAX jobs in United States

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

15 jobs
ElastixAI logoEL

Machine Learning Research Scientist

ElastixAI

Seattle, Washington, United States (Hybrid)

6 days ago
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