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

JAX jobs in United States

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

13 jobs
ElastixAI logoEL

Machine Learning Research Scientist

ElastixAI

Seattle, Washington, United States (Hybrid)

3 wk. ago
ML Research Scientist
AI Research Scientist
Machine Learning
PyTorch
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
AC

Machine Learning Research Engineer (MLRE) - Research

Achira

San Francisco, California, United States (Hybrid)

$164.6–259K Yearly4 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 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
Hugging Face logoHF

Open-Source Machine Learning Engineer - US Remote

Hugging Face

United States (Remote)

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

Senior Machine Learning Engineer, AI Generation Engine

SandboxAQ

United States (Remote)

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

Staff Machine Learning Engineer, AI Generation Engine

SandboxAQ

United States (Remote)

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