AI Jobs Requiring Jax

Find AI jobs requiring Jax skills. Browse 96 open positions.

96
Open Positions
$262K
Avg. Salary
11
Remote Roles

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

AI/ML Engineer
Staff Machine Learning Engineer, Generative AI Modeling and Inference
Snap Inc.
$195K - $343K Los Angeles, CA, US
View Role →
AI/ML Engineer
LLM Post-Training Engineer Graduate (Research & Product) - 2027 Start
TikTok
$128K - $316K San Jose, CA, US
View Role →
AI/ML Engineer
Founding Music AI Engineer
David Joseph & Company
$175K - $225K San Francisco, CA, US
View Role →
AI/ML Engineer
Senior Deep Learning Compiler Engineer - XLA
NVIDIA
$152K - $287K Austin, TX, US
View Role →
AI/ML Engineer
Senior Deep Learning Algorithm Engineer
NVIDIA
$184K - $356K Santa Clara, CA, US
View Role →
AI/ML Engineer
Senior AI Compiler Engineer, MLIR
NVIDIA
$152K - $287K Austin, TX, US
View Role →
AI/ML Engineer
Perception Machine Learning Engineer - Continuous Learning
Waymo
$175K - $215K Mountain View, CA, US
View Role →
AI Software Engineer
Software Engineer- AI/ML, Amazon Neuron Training
Amazon Web Services
$143K - $223K Cupertino, CA, US
View Role →
AI/ML Engineer
Staff Software Engineer, ML Systems, Accelerators
Google
$207K - $300K Sunnyvale, CA, US
View Role →
AI/ML Engineer
AI/ML Specialist Solutions Architect, Enterprise, AGS US Specialist SA
Amazon Web Services
$131K - $204K New York, NY, US
View Role →
AI/ML Engineer
ML Infrastructure Engineer
BV Teck
$100K - $150K Remote
View Role →
Research Scientist
Senior Research Scientist, Foundational AI in Health
Google
$174K - $252K Mountain View, CA, US
View Role →
AI/ML Engineer
Machine Learning Engineer, DesignX
Tesla
$140K - $258K Palo Alto, CA, US
View Role →
AI/ML Engineer
Software Engineer, On-Device Machine Learning
Google
$174K - $252K New York, NY, US
View Role →
AI/ML Engineer
Applied AI Scientist
Vantor
$128K - $215K Westminster, CO, US
View Role →
Data Scientist
Sr Applied Data Scientist - Personalization (applied ML, PyTorch, ML Ops)
Target
$98K - $176K Brooklyn Park, MN, US
View Role →
AI Software Engineer
Senior Software Engineer, AI/ML, Creative Intelligence
Google
$174K - $252K Mountain View, CA, US
View Role →
AI/ML Engineer
Explainable AI - Postdoctoral Researcher
Lawrence Livermore National Laboratory
Livermore, CA, US
View Role →
AI/ML Engineer
Senior Reinforcement Learning Engineer
Apptronik
Sunnyvale, CA, US
View Role →
AI/ML Engineer
AI Modeling Engineer
Palona AI
Los Altos, CA, US
View Role →
AI/ML Engineer
Sr. Reinforcement Learning & Autonomous Decision Systems Engineer
AUREX
$170K - $200K Huntsville, AL, US
View Role →
AI/ML Engineer
Machine Learning Engineer Graduate (E-Commerce Recommendation Mall) - 2027 Start (PhD)
TikTok
$153K - $300K Seattle, WA, US
View Role →
Research Engineer
Research Engineer
Goliath Partners
Remote
View Role →
AI/ML Engineer
Internship, Applied AI Engineer, AI Hardware (Fall 2026/Winter 2027)
Tesla
$100K - $150K Palo Alto, CA, US
View Role →
Research Scientist
Research Scientist – AI
BV Teck
$100K - $150K Remote
View Role →
AI/ML Engineer
Machine Learning Engineer - Health AIML
Apple
$205K - $374K Seattle, WA, US
View Role →
AI/ML Engineer
AI Infra Advisory Researcher
Lenovo
Morrisville, NC, US
View Role →
AI/ML Engineer
Senior Machine Learning Engineer, Agentic Science/Generative Models, AI for Biology & Translation (AIBT)
Genentech
$168K - $312K San Francisco, CA, US
View Role →
AI/ML Engineer
Senior Machine Learning Engineer, Perception
Smart Apply Test Company
Santa Clara, CA, US
View Role →
AI/ML Engineer
AI/ML Specialist Solutions Architect, Payments, AGS US Specialist SA
Amazon Web Services
$144K - $204K New York, NY, US
View Role →
AI/ML Engineer
Sr. AI/ML Specialist Solutions Architect, AGS Specialist Solutions Architects
Amazon Web Services
$153K - $207K New York, NY, US
View Role →
AI/ML Engineer
Senior Machine Learning Engineer, Perception
PlusAI
$150K - $250K Santa Clara, CA, US
View Role →
AI/ML Engineer
Engineering Manager, GenAI, Google Maps Immersive Navigation
Google
$207K - $300K Mountain View, CA, US
View Role →
AI/ML Engineer
Staff ML Engineer, Perception Research
Waymo
$251K - $310K Mountain View, CA, US
View Role →
Research Scientist
Senior Applied Scientist, Annapurna ML
Amazon Web Services
$167K - $260K Seattle, WA, US
View Role →
Research Scientist
Applied Scientist, Neuron ARG, Annapurna ML
Amazon Web Services
$142K - $222K Cupertino, CA, US
View Role →
AI/ML Engineer
ML Engineer, Foundation Model Infrastructure
Waymo
$175K - $215K Mountain View, CA, US
View Role →
AI/ML Engineer
Principal Software Quality Engineer – GPU & Machine Learning
AMD
San Jose, CA, US
View Role →
Research Scientist
Tech Lead Research Scientist/Engineer, Neural Graphics and World Models - TikTok
TikTok
$254K - $588K San Jose, CA, US
View Role →
Research Engineer
LLM Research Engineer IV
NuByt
$249K - $272K Mountain View, CA, US
View Role →
AI/ML Engineer
LLM Post-training Engineer Intern (Research & Product) - 2027 Summer
TikTok
San Jose, CA, US
View Role →
Research Engineer
AI Research Engineer
Normal Computing Corporation
$200K - $400K New York, NY, US
View Role →
AI/ML Engineer
AI Infrastructure Engineer
BV Teck
$100K - $160K Remote
View Role →
AI/ML Engineer
Software Engineer, CUDA Deep Learning Systems
NVIDIA
$124K - $195K Austin, TX, US
View Role →
AI/ML Engineer
Senior Software Engineer, CUDA Deep Learning Systems
NVIDIA
$184K - $356K Austin, TX, US
View Role →
AI/ML Engineer
Reinforcement Learning Engineer
Apptronik
Austin, TX, US
View Role →
Research Engineer
Principal AI/ML Research Engineer
WEX Inc.
$250K - $289K Remote
View Role →
Research Engineer
ML Software Research Engineer
Apple
Pittsburgh, PA, US
View Role →
AI/ML Engineer
Vice President, Artificial Intelligence
Enlyte
$232K - $250K Remote
View Role →
AI/ML Engineer
Machine Learning Senior Software Engineer
Akamai
$121K - $218K US
View Role →

Showing 50 of 96 jobs

About This Role

AI job market dashboard showing open roles by category

AI/ML Engineers build and deploy machine learning models in production. They work across the full ML lifecycle: data pipelines, model training, evaluation, and serving infrastructure. The role has evolved significantly over the past two years. Where ML Engineers once spent most of their time on model architecture, the job now tilts heavily toward inference optimization, cost management, and integrating LLM capabilities into existing systems. Companies want engineers who can ship production systems, and the experimenter-only role is fading fast.

Day-to-day, you're writing training pipelines, debugging data quality issues, setting up evaluation frameworks, and figuring out why your model performs differently in staging than it did on your dev set. The best ML engineers are obsessive about reproducibility and measurement. They instrument everything. They know that a model is only as good as the data feeding it and the infrastructure serving it.

Across the 4,317 AI roles we're tracking, AI/ML Engineer positions make up 70% of the market.

Demand for AI/ML Engineers has been strong and consistent. Unlike some AI roles that spike with hype cycles, ML engineering is a foundational need. Every company deploying AI models needs people who can keep them running, and the gap between research prototypes and production systems keeps growing.

Compensation Benchmarks

AI/ML Engineer roles pay a median of $214,900 based on 6,420 positions with disclosed compensation.

Across all AI roles, the market median is $215,000. Top-quartile compensation starts at $266,300. The 90th percentile reaches $320,790. For comparison, the highest-paying categories include AI Safety ($287,500) and Research Engineer ($272,100). By seniority level: Entry: $110,000; Mid: $194,400; Senior: $227,400; Director: $274,554; VP: $241,000.

AI Hiring Overview

The AI job market has 4,317 open positions tracked in our dataset. By seniority: 138 entry-level, 2,071 mid-level, 1,655 senior, and 453 leadership roles (Director, VP, C-Level). Remote roles make up 15% of the market (635 positions). The remaining 3,657 roles require on-site or hybrid attendance.

The market median for AI roles is $215,000. Top-quartile compensation starts at $266,300. The 90th percentile reaches $320,790. Highest-paying categories: AI Safety ($287,500 median, 34 roles); Research Engineer ($272,100 median, 227 roles); AI Engineering Manager ($244,000 median, 23 roles).

Demand for AI/ML Engineers has been strong and consistent. Unlike some AI roles that spike with hype cycles, ML engineering is a foundational need. Every company deploying AI models needs people who can keep them running, and the gap between research prototypes and production systems keeps growing.

Career Path

Common paths into AI/ML Engineer roles include Data Scientist, Software Engineer, Research Engineer.

From here, career progression typically leads toward ML Architect, AI Engineering Manager, Principal ML Engineer.

The fastest path into ML engineering is through software engineering with a self-directed ML education. A CS degree helps, but production engineering skills matter more than academic credentials. Build something that works, deploy it, and measure it. That portfolio project is worth more than a Coursera certificate. For career growth, the fork comes around the senior level: go deep on technical complexity (staff/principal track) or move into managing ML teams.

The AI Job Market Today

The AI job market spans 4,317 open positions across 15 role categories. The largest categories by volume: AI/ML Engineer (3,004), Data Scientist (345), AI Software Engineer (309). These three account for the majority of open positions, though smaller categories often have higher per-role compensation because of specialized skill requirements.

The seniority mix tells a story about where AI teams are in their maturity. Entry-level roles (138) are outnumbered by mid-level (2,071) and senior (1,655) positions, reflecting that most companies are past the 'build a team from scratch' phase and need experienced engineers who can ship production systems. Leadership roles (Director, VP, C-Level) total 453 positions, representing the bottleneck between technical execution and organizational strategy.

Remote work availability sits at 15% of all AI roles (635 positions), with 3,657 requiring on-site or hybrid attendance. The remote share has stabilized after the post-pandemic correction. Senior and specialized roles (Research Scientist, ML Architect) are more likely to be remote-eligible than entry-level positions, partly because experienced hires have more negotiating power and partly because these roles require less hands-on mentorship.

AI compensation is structured in clear tiers. The market median sits at $215,000. Top-quartile roles start at $266,300, and the 90th percentile reaches $320,790. These figures include base salary with disclosed compensation. Total compensation (including equity, bonuses, and sign-on) runs 20-40% higher at companies that offer those components.

Category matters for compensation. AI Safety roles lead at $287,500 median, while Prompt Engineer roles sit at $145,000. The spread between highest and lowest-paying categories reflects the premium on specialized technical skills versus broader analytical roles.

The most in-demand skills across all AI postings: Python (2,249 postings), Aws (1,224 postings), Azure (938 postings), Rag (915 postings), Gcp (660 postings), Pytorch (640 postings), Prompt Engineering (624 postings), Kubernetes (559 postings). Python dominates, appearing in the vast majority of role descriptions regardless of category. Cloud platform experience (AWS, GCP, Azure) is the second most common requirement. The newer entrants to the top skills list (RAG, vector databases, LLM APIs) reflect the shift from traditional ML toward generative AI applications.

Frequently Asked Questions

AI Pulse currently tracks 96 AI job openings that require Jax skills. 11 of these are remote positions.
AI roles requiring Jax pay an average of $262K 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.