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.

15 jobs
Mercor logoME

MLOps Specialist - Fully Remote

Mercor

Worldwide (Remote)

$35 – $45 Hourly1w ago
MLOps Engineering
ML Infrastructure Engineering
MLOps
JAX
Mercor logoME

Machine Learning Engineer | Upto $45/hr

Mercor

United States (Remote)

$35 – $45 Hourly1w ago
MLOps Engineer
ML Infrastructure Engineer
JAX
PyTorch
AC

ML Research Engineer - Training

Achira

San Francisco, California, United States (Hybrid)

$164.6K – $259K Yearly2w ago
machine-learning-research-engineer
gpu-optimization-engineer
pytorch
jax
AC

Machine Learning Research Engineer (MLRE) - Workflows/Systems

Achira

San Francisco, California, United States (Hybrid)

2w ago
Machine Learning
machine-learning-engineering
pytorch
jax
AC

Machine Learning Research Engineer (MLRE) - Research

Achira

San Francisco, California, United States (Hybrid)

2w 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)

2w ago
machine-learning-engineering
speech-ai
pytorch
jax
SandboxAQ logoSA

Staff ML Research Scientist, Co-Folding and Affinity

SandboxAQ

United States (Remote)

$204K – $306K Yearly2w ago
ml-research-scientist
ai-drug-discovery
Deep Learning
Machine Learning
Character.AI logoCH

Machine Learning Infrastructure Engineer

Character.AI

United States (Remote)

$150K – $350K Yearly2w ago
ml-infrastructure-engineering
machine-learning-infrastructure
gpu
Kubernetes
SandboxAQ logoSA

Senior Machine Learning Engineer, AI Generation Engine

SandboxAQ

United States (Remote)

$164K – $246K Yearly2w ago
machine-learning-engineering
ai-development
Python
pytorch
Bright Vision Technologies logoBV

AI Research Engineer (Applied AI)

Bright Vision Technologies

United States (Remote)

$100K – $150K Yearly2w ago
applied-ai-research
Machine Learning Engineering
Python
PyTorch
SandboxAQ logoSA

Staff Machine Learning Engineer, AI Generation Engine

SandboxAQ

United States (Remote)

$196K – $294K Yearly2w ago
machine-learning-engineer
AI Engineer
Python
pytorch
Hugging Face logoHF

Open-Source Machine Learning Engineer - US Remote

Hugging Face

United States (Remote)

2w ago
open-source
machine-learning-engineering
Python
pytorch
Abridge logoAB

Machine Learning Scientist (All Levels)

Abridge

San Francisco, California, United States (Hybrid)

$205K – $300K Yearly2w ago
machine-learning-engineer
Research Scientist
Machine Learning
natural-language-processing
Bright Vision Technologies logoBV

GPU Systems Engineer (CUDA)

Bright Vision Technologies

United States (Remote)

$100K – $150K Yearly1w ago
GPU Systems Engineer
CUDA Developer
CUDA
C++
Zoox logoZO

Machine Learning Engineer - Semantic Reasoning (Highway)

Zoox

Foster City, California, United States (Hybrid)

$189K – $258K Yearly2w ago
Machine Learning Engineering
autonomous-driving
Python
C++