Inference Jobs logo

Inference Jobs

  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 Inference Jobs and apply today.

2mo agoXA

Member of Technical Staff, Midtraining

xAI

Palo Alto, California, United States (On-site)$180K – $440K Yearly
Machine Learning Engineer
Data Engineer
Python
JAX
3mo agoCO

Senior Member of Technical Staff, Multimodal AI

Cohere

San Francisco, California, United States or Remote (Worldwide)
Machine Learning Engineer
AI Research Engineer
Python
JAX
3mo agoAN

Research Engineer, Audio

Anthropic

San Francisco, California, United States (Hybrid)$350K – $500K Yearly
AI Research & Engineering
Audio Engineering
JAX
PyTorch
4w agoLA

Applied Research Engineer, Agents

Labelbox

San Francisco, California, United States (Hybrid)$250K – $300K Yearly
AI Research
Machine Learning Engineering
Python
PyTorch
3w agoTM

Research Engineer, Infrastructure, Training Systems

Thinking Machines Lab

San Francisco, California, United States (On-site)$350K – $475K Yearly
Machine Learning Infrastructure
Research Infrastructure (ML Infra)
PyTorch
JAX
3mo agoCO

Member of Technical Staff, Modeling

Cohere

London, England, United Kingdom or Remote (Worldwide)
Modeling
AI Research
Python
TensorFlow
3w agoTM

Research Engineer, Infrastructure, Tinker

Thinking Machines Lab

San Francisco, California, United States (On-site)$350K – $475K Yearly
ML Infrastructure
Machine Learning Engineer
Deep Learning Frameworks
PyTorch
3w agoTM

Research, Pre-Training Data

Thinking Machines Lab

San Francisco, California, United States (On-site)$350K – $475K Yearly
Data Engineering
Data Science
Data Engineering
JAX
3w agoTM

Research, Audio Expertise

Thinking Machines Lab

San Francisco, California, United States (On-site)$350K – $475K Yearly
Machine Learning Research
AI Research
Python
PyTorch
3w agoTM

Research Engineer, Infrastructure, Numerics

Thinking Machines Lab

San Francisco, California, United States (On-site)$350K – $475K Yearly
Research Infrastructure (ML Infra)
Machine Learning Infrastructure
PyTorch
JAX
4w agoSC

Machine Learning Research Scientist/ Engineer, Agents

Scale

San Francisco, California, United States (On-site)$275K – $350K Yearly
Machine Learning Research
Machine Learning Engineer
LLMs
PyTorch
2w agoAN

Research Engineer, Pretraining Scaling

Anthropic

San Francisco, California, United States (On-site)$350K – $850K Yearly
Research Engineer
Machine Learning Engineer
JAX
TPU
4w agoXA

Member of Technical Staff - Multimodal

xAI

Palo Alto, California, United States (On-site)$180K – $440K Yearly
Machine Learning Engineer
AI Research
Python
PyTorch
2mo agoDE

Staff Software Engineer, ML Infrastructure

Decagon

San Francisco, California, United States (On-site)$300K – $430K Yearly
Engineering
Infrastructure Engineering
Python
PyTorch
3w agoTM

Research Engineer, Infrastructure, Inference

Thinking Machines Lab

San Francisco, California, United States (On-site)$350K – $475K Yearly
Machine Learning Infrastructure
AI Infrastructure
PyTorch
JAX
1d agoNV

Senior Solutions Architect, GPU - Cloud Service Providers

NVIDIA

Santa Clara, California, United States (On-site)$184K – $356.5K Yearly
Solutions Architecture
Cloud Infrastructure
PyTorch
JAX
2mo agoXA

Member of Technical Staff, Model Evaluation

xAI

Palo Alto, California, United States (On-site)$180K – $440K Yearly
Machine Learning Engineer
Model Evaluation
Python
JAX
3w agoTM

Research, Pre-Training Science

Thinking Machines Lab

San Francisco, California, United States (On-site)$350K – $475K Yearly
Research Scientist
Machine Learning Research
Python
PyTorch