Get AI Careers logo
Get AI Careers
  1. Home
  2. Jobs
  3. United States
  4. California
  5. JAX

JAX jobs in California, United States

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

9 jobs
Gauss Labs logoGL

AI Scientist - Machine Learning (US/KR)

Gauss Labs

Palo Alto, California, United States (Hybrid)

1 yr. ago
AI Research
Machine Learning
Machine Learning
Deep Learning
AC

Machine Learning Research Engineer (MLRE) - Workflows/Systems

Achira

San Francisco, California, United States (Hybrid)

$164.6–259K Yearly4 mo. ago
Machine Learning
Machine Learning Engineering
PyTorch
JAX
Character.AI logoCH

Machine Learning Infrastructure Engineer

Character.AI

Redwood City, California, United States (Hybrid)

$150–350K Yearly8 mo. ago
ML Infrastructure Engineering
Machine Learning Infrastructure
GPU
Kubernetes
Abridge logoAB

Machine Learning Scientist (All Levels)

Abridge

San Francisco, California, United States (Hybrid)

$205–300K Yearly8 mo. ago
Machine Learning Scientist
Research Scientist
Machine Learning
Natural Language Processing
Plaud logoPL

Machine Learning Engineer, Speech LLM Training - San Francisco

Plaud

San Francisco, California, United States (Hybrid)

$195–365K Yearly4 mo. ago
Machine Learning Engineering
Speech AI
PyTorch
JAX
Zillow Group logoZG

Machine Learning Engineer

Zillow Group

United States (Remote)

4 mo. ago
Machine Learning Engineer
ML Engineering
Python
PyTorch
SandboxAQ logoSA

Staff Machine Learning Engineer, AI Generation Engine

SandboxAQ

United States (Remote)

$196–294K Yearly3 wk. ago
Machine Learning Engineer
AI Engineer
Python
PyTorch
Centific logoCE

Research Scientist, LLM Evaluation & Post-Training

Centific

Palo Alto, California, US or Remote (United States)

$150–300K Yearly5 mo. ago
Research Scientist
LLM Evaluation
Python
PyTorch