Get AI Careers logo
Get AI Careers
  1. Home
  2. Jobs
  3. JAX

JAX jobs

Explore JAX roles on Get AI Careers and apply today.

15 jobs
ElastixAI logoEL

Machine Learning Research Scientist

ElastixAI

Seattle, Washington, United States (Hybrid)

5 days ago
ML Research Scientist
AI Research Scientist
Machine Learning
PyTorch
Zillow Group logoZG

Machine Learning Engineer

Zillow Group

United States (Remote)

4 wk. ago
Machine Learning Engineer
ML Engineering
Python
PyTorch
Hugging Face logoHF

Open-Source Machine Learning Engineer - US Remote

Hugging Face

United States (Remote)

4 days ago
Open Source
Machine Learning Engineering
Python
PyTorch
Gauss Labs logoGL

AI Scientist - Machine Learning (US/KR)

Gauss Labs

Palo Alto, California, United States (Hybrid)

2 wk. ago
AI Research
Machine Learning
Machine Learning
Deep Learning
AC

Machine Learning Research Engineer (MLRE) - Workflows/Systems

Achira

San Francisco, California, United States (Hybrid)

4 days ago
Machine Learning
Machine Learning Engineering
PyTorch
JAX
Character.AI logoCH

Machine Learning Infrastructure Engineer

Character.AI

United States (Remote)

$150–350K Yearly4 days 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 Yearly4 days ago
Machine Learning Engineer
Research Scientist
Machine Learning
Natural Language Processing
SandboxAQ logoSA

Senior Machine Learning Engineer, AI Generation Engine

SandboxAQ

United States (Remote)

$164–246K Yearly4 days ago
Machine Learning Engineering
AI Development
Python
PyTorch
Zoox logoZO

Senior Engineering Manager, ML Platform

Zoox

Foster City, California, United States (Hybrid)

$317–370K Yearly2 wk. ago
Machine Learning Infrastructure Engineering
Engineering Management
PyTorch
JAX
SandboxAQ logoSA

Staff Machine Learning Engineer, AI Generation Engine

SandboxAQ

United States (Remote)

$196–294K Yearly4 days ago
Machine Learning Engineer
AI Engineer
Python
PyTorch
Centific logoCE

Research Scientist, LLM Evaluation & Post-Training

Centific

Palo Alto, California, United States (Hybrid)

$150–300K Yearly2 wk. ago
Research Scientist
LLM Evaluation
LLM Evaluation
Post-training
Plaud logoPL

Machine Learning Engineer, Speech LLM Training - San Francisco

Plaud

San Francisco, California, United States (Hybrid)

4 days ago
Machine Learning Engineering
Speech AI
PyTorch
JAX
AC

Machine Learning Research Engineer (MLRE) - Research

Achira

San Francisco, California, United States (Hybrid)

4 days ago
Machine Learning Research Engineer
ML Research Scientist
PyTorch
JAX
Zoox logoZO

Machine Learning Engineer - Semantic Reasoning (Highway)

Zoox

Foster City, California, United States (Hybrid)

$189–258K Yearly3 days ago
Machine Learning Engineering
Autonomous Driving
Python
C++