Join Community Newsletter

AI Jobs Requiring JAX

Find AI jobs requiring JAX skills. Browse 70 open positions.

70
Open Positions
$246K
Avg. Salary
7
Remote Roles

Data updated weekly. Last refreshed 2026-03-08.

AI/ML Engineer
Machine Learning Engineer (Contractor)
Tech Holding
Remote
View Role →
Prompt Engineer
Research Software Engineer, Multimodal AI
Google
$166K - $244K San Jose, CA, US
View Role →
AI/ML Engineer
ML Engineer, Foundation Model Infrastructure
Waymo
$204K - $259K Remote
View Role →
AI/ML Engineer
Early Career AI/ML Engineer
BRAIN
San Francisco, CA, US
View Role →
Research Engineer
Sr. Research Engineer
Adobe
$164K - $313K Seattle, WA, US
View Role →
LLM Engineer
Machine Learning - Compiler Engineer II, AWS Neuron, Annapurna Labs
Amazon Web Services
$129K - $223K Seattle, WA, US
View Role →
LLM Engineer
Machine Learning - Compiler Engineer II, AWS Neuron, Annapurna Labs
Amazon Web Services
$129K - $223K Cupertino, CA, US
View Role →
AI/ML Engineer
Sr. Machine Learning - Compiler Engineer III, AWS Neuron, Annapurna Labs
Amazon Web Services
$151K - $261K Seattle, WA, US
View Role →
AI/ML Engineer
Sr. Machine Learning - Compiler Engineer III, AWS Neuron, Annapurna Labs
Amazon Web Services
$151K - $261K Cupertino, CA, US
View Role →
AI/ML Engineer
Software Engineer- AI/ML, AWS Neuron Distributed Training
Amazon Web Services
$129K - $223K Seattle, WA, US
View Role →
AI/ML Engineer
Software Engineer- AI/ML, AWS Neuron Distributed Training
Amazon Web Services
$129K - $223K Cupertino, CA, US
View Role →
AI/ML Engineer
Sr. Software Engineer- AI/ML, AWS Neuron Distributed Training
Amazon Web Services
$151K - $261K Seattle, WA, US
View Role →
AI/ML Engineer
Sr. Software Engineer- AI/ML, AWS Neuron Distributed Training
Amazon Web Services
$151K - $261K Cupertino, CA, US
View Role →
AI/ML Engineer
Software Development Engineer - AI/ML, AWS Neuron, Multimodal Inference
Amazon Web Services
$129K - $223K Cupertino, CA, US
View Role →
AI/ML Engineer
Software Development Engineer - AI/ML, AWS Neuron, Multimodal Inference
Amazon Web Services
$129K - $223K Seattle, WA, US
View Role →
AI/ML Engineer
Senior Software Development Engineer - AI/ML, AWS Neuron, Multimodal Inference
Amazon.com
$151K - $261K Cupertino, CA, US
View Role →
AI/ML Engineer
Senior Software Development Engineer - AI/ML, AWS Neuron, Multimodal Inference
Amazon.com
$151K - $261K Seattle, WA, US
View Role →
AI/ML Engineer
Senior Software Development Engineer, AI/ML, AWS Neuron, Model Inference
Amazon Web Services
$151K - $261K Cupertino, CA, US
View Role →
AI/ML Engineer
Senior Software Development Engineer, AI/ML, AWS Neuron, Model Inference
Amazon Web Services
$151K - $261K Seattle, WA, US
View Role →
AI/ML Engineer
Software Engineer III, Generative AI/ML Frameworks
Google
$141K - $202K Mountain View, CA, US
View Role →
AI/ML Engineer
Advisor - Agent Research
Eli Lilly
$151K - $244K Indianapolis, IN, US
View Role →
AI/ML Engineer
Staff Software Engineer – AI (Westlaw AI)
Thomson Reuters
$127K - $236K Eagan, MN, US
View Role →
Research Engineer
Sr. Research Engineer, Machine Learning, AGI Foundations
Amazon.com
$168K - $250K Bellevue, WA, US
View Role →
Research Engineer
Sr. Research Engineer, Machine Learning, AGI Foundations
Amazon.com
$168K - $250K New York, NY, US
View Role →
Research Engineer
Sr. Research Engineer, Machine Learning, AGI Foundations
Amazon.com
$168K - $250K Boston, MA, US
View Role →
Research Engineer
Sr. Research Engineer, Machine Learning, AGI Foundations
Amazon.com
$168K - $250K Sunnyvale, CA, US
View Role →
Research Scientist
Applied Scientist, ML Compiler
Amazon Web Services
$136K - $223K Cupertino, CA, US
View Role →
Research Scientist
Applied Scientist, ML Compiler
Amazon Web Services
$136K - $223K Seattle, WA, US
View Role →
Data Scientist
Senior Data Scientist, Data and Machine Learning, WWPS US Federal
Amazon Web Services
$159K - $215K Arlington, VA, US
View Role →
Data Scientist
Senior Data Scientist, Data and Machine Learning, WWPS US Federal
Amazon Web Services
$159K - $215K Jessup, MD, US
View Role →
Data Scientist
Senior Data Scientist, Data and Machine Learning, WWPS US Federal
Amazon Web Services
$159K - $215K Denver, CO, US
View Role →
Data Scientist
Senior Data Scientist, Data and Machine Learning, WWPS US Federal
Amazon Web Services
$159K - $215K Herndon, VA, US
View Role →
Data Scientist
Senior Data Scientist, Data and Machine Learning, WWPS US Federal
Amazon Web Services
$159K - $215K St. Louis, MO, US
View Role →
Data Scientist
Data Scientist, Data and Machine Learning, WWPS US Federal
Amazon Web Services
$136K - $184K Denver, CO, US
View Role →
Data Scientist
Data Scientist, Data and Machine Learning, WWPS US Federal
Amazon Web Services
$136K - $184K Arlington, VA, US
View Role →
Data Scientist
Data Scientist, Data and Machine Learning, WWPS US Federal
Amazon Web Services
$136K - $184K St. Louis, MO, US
View Role →
Data Scientist
Data Scientist, Data and Machine Learning, WWPS US Federal
Amazon Web Services
$136K - $184K Herndon, VA, US
View Role →
Data Scientist
Data Scientist, Data and Machine Learning, WWPS US Federal
Amazon Web Services
$136K - $184K Jessup, MD, US
View Role →
AI/ML Engineer
2026 Applied Science Internship - United States, PhD Student Science Recruiting, Frontier AI & Robotics
Amazon.com
$135K - $222K San Francisco, CA, US
View Role →
AI/ML Engineer
2026 Applied Science Internship - United States, PhD Student Science Recruiting, Frontier AI & Robotics
Amazon.com
$135K - $222K Seattle, WA, US
View Role →
AI/ML Engineer
Sr. Staff AI/ML Development Engineer
AMD
Remote
View Role →
AI/ML Engineer
Sr. Staff AI/ML Development Engineer
AMD
$192K - $288K Remote
View Role →
AI/ML Engineer
Staff Machine Learning Engineer (Modeling), Support
Block
$276K - $415K Remote
View Role →
Prompt Engineer
Lead Security Engineer
JPMorganChase
Plano, TX, US
View Role →
AI/ML Engineer
2026 PhD Artificial Intelligence Models Software Engineer Co-Op/Intern
AMD
San Jose, CA, US
View Role →
AI/ML Engineer
Principal AI-Machine Learning Engineer
Dana-Farber Cancer Institute
$154K - $184K Boston, MA, US
View Role →
AI/ML Engineer
AI-Machine Learning Engineer
Dana-Farber Cancer Institute
$90K - $106K Boston, MA, US
View Role →
AI/ML Engineer
ML Engineer, Foundation Model Evaluation
Waymo
$170K - $216K Remote
View Role →
AI/ML Engineer
Senior Machine Learning Engineer - Autonomy Validation
Zoox
$229K - $276K Foster City, CA, US
View Role →
AI/ML Engineer
Machine Learning Engineer, AWS Neuron Inference, AWS Neuron
Amazon Web Services
$143K - $194K Cupertino, CA, US
View Role →

Showing 50 of 70 jobs

Frequently Asked Questions

AI Pulse currently tracks 70 AI job openings that require JAX skills. 7 of these are remote positions.
AI roles requiring JAX pay an average of $246K based on disclosed compensation. Specialized skills like JAX combined with production experience typically command 10-20% premiums over general AI roles.

Get Weekly AI Career Intelligence

Salary data, skills demand, and market signals from 16,000+ AI job postings. Every Monday.

Browse Jobs by Skill

Python AIPyTorch AITensorFlow AILangChain AIRAG AIAWS AILLM EngineerFine-Tuning AI