Inference Jobs logo

Inference Jobs

  1. Home
  2. Jobs
  3. United States
  4. California
  5. San Francisco
  6. JAX

JAX Jobs in San Francisco, California, United States

Discover JAX roles in San Francisco, California, United States on Inference Jobs and apply today.

2w agoAN

Research Engineer, Discovery

Anthropic

San Francisco, California, United States (Hybrid)$350K – $850K Yearly
Machine Learning Infrastructure
Research Engineer
Distributed Systems
Docker
3mo agoAC

Research Engineer

Applied Compute

San Francisco, California, United States (On-site)
Research Engineer
Machine Learning Engineer
Large Language Models
Reinforcement Learning
3mo agoOP

Researcher, Pretraining Safety

OpenAI

San Francisco, California, United States (On-site)$310K – $460K Yearly
AI Research
AI Safety
Python
PyTorch
2mo agoTM

Research Infrastructure Engineer, Research Acceleration

Thinking Machines Lab

San Francisco, California, United States (On-site)$350K – $475K Yearly
Research Infrastructure (ML Infra)
Machine Learning Engineer
Python
Rust
3w agoTM

Software Engineer, Supercomputing

Thinking Machines Lab

San Francisco, California, United States (On-site)$350K – $475K Yearly
Software Engineering
Infrastructure Engineering
Python
Rust
2w agoAN

Research Engineer, Interpretability

Anthropic

San Francisco, California, United States (Hybrid)$315K – $560K Yearly
AI Research
Machine Learning Engineer
Python
Rust
3mo agoPO

Member of Engineering (Pre-training and inference software)

Poolside

United Kingdom or Remote (Europe, Middle East, and Africa, North America)
Machine Learning Engineer
LLM Training
Python
PyTorch
3w agoTM

Research Engineer, Infrastructure, Kernels

Thinking Machines Lab

San Francisco, California, United States (On-site)$350K – $475K Yearly
Machine Learning Infrastructure
AI Research Engineer
CUDA
CuTe
2mo agoTM

Software Engineer, Research Acceleration

Thinking Machines Lab

San Francisco, California, United States (On-site)$350K – $475K Yearly
Software Engineer
ML Infrastructure
Python
Rust
2w agoAN

Privacy Research Engineer, Safeguards

Anthropic

San Francisco, California, United States (Hybrid)$320K – $485K Yearly
Privacy Engineering
Machine Learning Engineering
Privacy-Preserving Machine Learning
Python
2w agoAN

Research Engineer, Machine Learning (Horizons)

Anthropic

San Francisco, California, United States (Hybrid)$500K – $850K Yearly
Machine Learning Engineering
AI Research
Python
Reinforcement Learning
2w agoAN

ML Infrastructure Engineer, Safeguards

Anthropic

San Francisco, California, United States (Hybrid)$320K – $405K Yearly
Machine Learning Engineer
MLOps
Python
PyTorch
4w agoLA

Applied Research Intern

Labelbox

San Francisco, California, United States (Hybrid)$35 – $45 Hourly
Applied Research
Machine Learning Research
Python
PyTorch
3mo agoPE

AI Engineer, Applied ML

Perplexity

San Francisco, California, United States (On-site)$210K – $385K Yearly
Machine Learning Engineer
AI Engineer
Machine Learning
LLMs
3mo agoOP

Research Engineer, Privacy

OpenAI

San Francisco, California, United States (On-site)$380K – $460K Yearly
Privacy Engineering
Security
Privacy-Enhancing Technologies
Differential Privacy
2w agoAN

Performance Engineer, GPU

Anthropic

San Francisco, California, United States (Hybrid)$280K – $850K Yearly
Performance Engineer
GPU Engineer
CUDA
GPU Programming