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.

12 jobs
d-Matrix logoD-

Senior Staff ML Researcher - LLM Algorithmic Optimization

d-Matrix

Santa Clara, California, United States (Hybrid)

$175–265K Yearly4 mo. ago
ML Research Scientist
LLM Research
Python
Machine Learning
Gauss Labs logoGL

AI Scientist - Machine Learning (US/KR)

Gauss Labs

Palo Alto, California, United States (Hybrid)

2 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 Yearly6 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 Yearly2 yr. 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 Yearly1 yr. ago
Machine Learning Scientist
Research Scientist
Machine Learning
Natural Language Processing
AC

Machine Learning Research Engineer (MLRE) - Research

Achira

San Francisco, California, United States (Hybrid)

$164.6–259K Yearly6 mo. ago
Machine Learning Research Engineer
ML Research Scientist
PyTorch
JAX
Plaud logoPL

Machine Learning Engineer, Speech LLM Training - San Francisco

Plaud

San Francisco, California, United States (Hybrid)

$195–365K Yearly6 mo. ago
Machine Learning Engineering
Speech AI
PyTorch
JAX
OpenTrain AI logoOA

Senior Machine Learning Engineer, Part-Time AI Model Work - Remote (United States)

OpenTrain AI

United States (Remote)

$60.00 – $120.00 Hourly5 days ago
Machine Learning Engineer
AI Engineer
Machine Learning
Python
Hugging Face logoHF

Open-Source Machine Learning Engineer - US Remote

Hugging Face

United States (Remote)

5 mo. ago
Open Source
Machine Learning Engineering
Python
PyTorch
SandboxAQ logoSA

Senior Machine Learning Engineer, AI Generation Engine

SandboxAQ

United States (Remote)

$164–246K Yearly6 mo. ago
Machine Learning Engineering
AI Development
Python
PyTorch
SandboxAQ logoSA

Staff Machine Learning Engineer, AI Generation Engine

SandboxAQ

United States (Remote)

$196–294K Yearly10 mo. 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 Yearly7 mo. ago
Research Scientist
LLM Evaluation
Python
PyTorch