Skip to content

JobsBest AI CompaniesBest Companies

Updated 19h ago

AI Job Signal

A curated board of AI, ML, and LLM engineering roles, for the people who build AI systems. AI, ML & LLM engineering roles.

Looking for worldwide remote roles? globalremotely →

Roles

AI Engineer Machine Learning Engineer AI Agent Engineer Forward Deployed Engineer Applied Scientist Computer Vision Engineer AI Research All roles →

Skills

Python PyTorch TensorFlow AWS Kubernetes Go CUDA All skills →

Locations

United States Europe Asia United Kingdom Canada Latin America All locations →

AI Job Signal · AI, ML & LLM engineering jobs.

New today Companies Best companies YC companies About How we source jobs Contact Privacy

JAX Jobs · 75 Developer & Engineering Roles · AI Job Signal

Home/Skills/JAX Jobs · 76 Developer & Engineering Roles

JAX Jobs · 76 Developer & Engineering Roles

2027 Summer Intern, PhD, Machine Learning, Simulation Realism

waymo · Internship · Python, PyTorch, TensorFlow

United Kingdom3w ago

2027 Summer Intern, MS/PhD, Machine Learning, Planning/Prediction

waymo · Internship · Python, PyTorch, TensorFlow

California (CA)3w ago

Machine Learning Engineer, Applied AI Co-Op Intern, Winter 2027

Brain Co. · Internship · Python, PyTorch, TensorFlow

California (CA)

4w ago

Machine Learning Engineer (Physics-based)

Root Access · Full-time · Python, PyTorch, JAX

New York (NY)4w ago

Member of Technical Staff, LLM & Production

Hark · Full-time · PyTorch, JAX, SGLang

United States4w ago

Research Engineer, Safety & Alignment

Cognition · Full-time · Python, PyTorch, JAX

California (CA)4w ago

2027 Summer Intern, PhD, Machine Learning, Planning/Prediction

waymo · Internship · Python, PyTorch, TensorFlow

California (CA)4w ago

Research Scientist Intern, AI Molecular Design (Winter/Spring)

GenBio AI · Internship · PyTorch, TensorFlow, JAX

California (CA)4w ago

Senior Software Engineer, Machine Learning Platform

Airwallex · Full-time · Python, PyTorch, TensorFlow

Singapore4w ago

PhD Machine Learning Software Engineer Intern (Summer 2027)

lyft · Internship · Python, PyTorch, TensorFlow

California (CA)4w ago

Machine Learning Engineer — Reinforcement Learning

Institute of Foundation Models · Full-time · PyTorch, Kubernetes, DeepSpeed

California (CA)1mo ago

Machine Learning Engineer — Pre-training (LLM)

Institute of Foundation Models · Full-time · Python, PyTorch, Kubernetes

California (CA)1mo ago

Senior Machine Learning Scientist

Flagship Pioneering, Inc. · Full-time · Python, PyTorch, JAX

United Kingdom1mo ago

2027 Summer Intern, MS/PhD, Machine Learning, Driver Refinement Foundations

waymo · Internship · Python, SQL, PyTorch

California (CA)1mo ago

Research Engineer, Post-training

Medra · Full-time · Python, PyTorch, JAX

California (CA)1mo ago

Research Engineer, ML Infrastructure

Chai Discovery · Full-time · Python, PyTorch, CUDA

California (CA)1mo ago