⚡

Jax Jobs

27 positions requiring Jax

Browse tech jobs that require Jax skills. From startups to Fortune 500 companies, find your next role in software engineering, data science, and more.

Open Positions

500+

Companies

Daily

Updated

🌍

Worldwide

Latest Jax Jobs

Senior Software Engineer, Machine Learning, Kernel

Google • Sunnyvale, CA, USA, Kirkland, WA, USA

C++Python PyTorch Jax

1w ago

Senior Staff Software Engineer, Torch TPU

Google • Sunnyvale, CA, USA

Python C++PyTorch Jax

1w ago

Software Engineer, Search AI, Search Quality

Google • Mountain View, CA, USA

Python C++Natural language processing Machine learning Deep learning Jax+1

1w ago

Machine Learning Engineer, AWS Neuron Inference, Annapurna ML

Amazon • Seattle, Washington, USA

AWS Python PyTorch Jax

1w ago

2026 Summer Intern, PhD, ML Training Infra and Efficiency

Waymo • Mountain View, CA, USA

Python C++Reinforcement learning TensorFlow Jax PyTorch

2w ago

Customer Engineer, AI, Infrastructure Modernization, Google Cloud (Japanese, English)

Google • Tokyo, Japan

GCP PyTorch TensorFlow Jax Ray Machine learning+1

2w ago

AI Incubation Lead, VertexAI, Core, Global Services Delivery

Google • Austin, TX, USA, New York, NY, USA, Reston, VA, USA, Seattle, WA, USA, Washington D.C., DC, USA, Miami, FL, USA

Vllm Slurm Pallas Jax Machine learning Recommendation systems+4

2w ago

Software Development Manager, AWS Neuron Machine Learning Distributed Training - Model Enablement

Amazon • Cupertino, California, USA

AWS PyTorch Jax Xla Fsdp Ddp

2w ago

AI/ML Engineer, Google Public Sector

Google • Reston, VA, USA

Python PyTorch TensorFlow Jax Ray C+++3

2w ago

Software Development Manager, AWS Neuron Machine Learning Distributed Training - Model Enablement

Amazon • Cupertino, California, USA

AWS PyTorch Jax Xla Fsdp Ddp

2w ago

Research Software Engineer, Multimodal AI

Google • San Jose, CA, USA

C++Python Generative ai Machine learning Jax TensorFlow+1

3w ago

Software Engineer III, AI/ML, Shopping Targeting

Google • Mountain View, CA, USA

C++Machine learning Deep learning Ml infrastructure Adsml PyTorch+3

3w ago

Software Development Manager, ML Accelerators, AWS Neuron, Annapurna Labs

Amazon • Seattle, Washington, USA

AWS Machine learning PyTorch TensorFlow Jax

3w ago

Post-Doctoral Researcher, PhD, 2026 Start (Fixed-Term Contract)

Google • Bengaluru, Karnataka, India

Python C++Java Jax TensorFlow PyTorch

3w ago

Software Development Engineer - AI/ML, AWS Neuron, Multimodal Inference

Amazon • Cupertino, California, USA

AWS PyTorch Jax Machine learning Deep learning

3w ago

Senior Software Development Engineer - AI/ML, AWS Neuron, Multimodal Inference

Amazon • Cupertino, California, USA

AWS PyTorch Jax Machine learning Deep learning

3w ago

2026 Annapurna Labs at AWS Internships (US) - Machine Learning Systems & Silicon Innovation

Amazon • Cupertino, California, USA

AWS PyTorch Jax Compiler development Distributed training systems Ml infrastructure development+2

1 month ago

Senior ML/RL Training Infrastructure Engineer

Apple • Zurich, Zurich, Switzerland

Python TensorFlow Jax Kubernetes MLflow Docker

1 month ago

Senior Software Engineer, AI Fitness, Health Coach, Fitbit

Google • London, UK

Python C++TensorFlow PyTorch Jax

1 month ago

Software Development Manager , AWS Neuron, Annapurna Labs

Amazon • Seattle, Washington, USA

AWS Machine learning PyTorch TensorFlow Jax

1 month ago

Software Engineer- AI/ML, AWS Neuron

Amazon • Cupertino, California, USA

AWS Python PyTorch Jax TensorFlow Deepspeed+1

1 month ago

Software Development Engineer, Distributed Systems, Annapurna Labs

Amazon • Seattle, Washington, USA

AWS Python TensorFlow PyTorch Jax Machine learning

1 month ago

Software Development Engineer, AI/ML, AWS Neuron, Model Inference

Amazon • Cupertino, California, USA

AWS PyTorch Jax Machine learning Deep learning

1 month ago

Software Development Engineer, AI/ML, AWS Neuron, Model Inference

Amazon • Cupertino, California, USA

AWS PyTorch Jax Machine learning Deep learning

1 month ago

Sr. Software Engineer - Data Engineering, Siri Speech

Apple • Cambridge, Massachusetts, United States

Pyspark Jax Ray Machine learning Natural language processing

1 month ago

Want to see all Jax jobs?

Unlock All Jobs →