Inference Jobs logo

Inference Jobs

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

JAX Jobs in Sunnyvale, California, United States

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

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 agoCO

Member of Technical Staff, Modeling

Cohere

London, England, United Kingdom or Remote (Worldwide)
Modeling
AI Research
Python
TensorFlow
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
4w agoAI

ML Runtime Optimization Engineer

Applied Intuition

Sunnyvale, California, United States (On-site)$159.1K – $199.3K Yearly
Machine Learning Engineering
Embedded Software Engineering
PyTorch
JAX
3mo agoPO

Member of Engineering (Scalability)

Poolside

United Kingdom or Remote (Europe, Middle East, and Africa, North America)
R&D
Machine Learning Infrastructure
Linux Kernel
PyTorch
3mo agoPO

Member of Engineering (Pre-training / Data)

Poolside

United Kingdom or Remote (Europe, Middle East, and Africa, North America)
Machine Learning Engineer
AI Researcher
Machine Learning
Large Language Models (LLMs)
3mo agoOP

Senior Full-Stack Engineer

OpenRouter

United States or Remote (United States)
Full Stack Development
Software Engineering
React
TypeScript