What is JAX and how common is it in AI job postings?

JAX is a skill in the "Frameworks & Tools" category. It currently appears in 242 active AI roles across 39 companies in our index.

Which companies are hiring for JAX?

The top employers with active AI roles mentioning JAX are: NVIDIA (51), Amazon (38), Google (34), Apple (30), Cohere (15).

Is demand for JAX growing?

Over the last 12 weeks, 138 new AI postings mentioned JAX. Demand is rising — up 139% in the last four weeks compared to the earliest four weeks in the window.

At which stage of AI development is JAX used?

Roles requiring JAX are concentrated in: serving infrastructure (40%), post-training (17%), agents (14%). These stages follow a seven-stage AI lifecycle from data preparation through to shipped product.

What skills are commonly paired with JAX?

Job postings that mention JAX most often also require: PyTorch, Python, Machine Learning, TensorFlow, Computer Architecture.

← All skills

JAX

242 active AI roles across 39 companies mention JAX. Category: Frameworks & Tools.

Demand trend

New AI postings mentioning JAX per week — 138 total over 12 weeks.

Sector

All Big Tech · 116 Semiconductors · 59 AI Frontier · 42 Robotics · 7 Enterprise · 4 Data AI · 3 Vertical AI · 2 Multimodal · 2 Fintech · 2 Consumer · 2 Retail · 1 Insurance · 1 Defense · 1

Function

All Engineering · 179 Research · 63

Status

All Active only

Sort

AI score Recently posted Company A–Z

FilteredfunctionResearch×

63 AI roles requiring this skill.

Company	Title	Sector	AI score	Stage
Google	Post-Doctoral Researcher, PhD, 2026 Start, DeepMind (Fixed-Term Contract) -	Big Tech	10	L1
OpenAI	Researcher, Training - London	AI Frontier	10	L1
Meta	Postdoctoral Researcher (PhD) - FAIR - Core Learning	Big Tech	10	L1
NVIDIA	Senior Research Scientist, Multimodal Foundation Models and Robotics	Semiconductors	10	L2
Mistral AI	Research Engineer, Machine Learning - Paris/London/Zurich/Warsaw	AI Frontier	10	L1
Mistral AI	AI Scientist - Zurich	AI Frontier	10	L1
Mistral AI	AI Scientist - Audio	AI Frontier	10	L1
Mistral AI	AI Scientist - Paris/London - Onsite or Hybrid or Remote	AI Frontier	10	L1
Mistral AI	AI Scientist - Palo Alto	AI Frontier	10	L1
NVIDIA	Research Scientist, Generalist Embodied Agent Research - PhD New College Grad 2026	Semiconductors	10	L4
NVIDIA	Senior Research Scientist, Multimodal Foundation Models and Robotics	Semiconductors	10	L2
Google	Research Engineer, Frontier Safety Mitigations, DeepMind	Big Tech	9	L4
Google	Research Engineer, Frontier Safety Mitigations, DeepMind	Big Tech	9	L4
Apple	Principal Research Scientist, Siri Innovation Studio	Big Tech	9	L6
Apple	AIML - Machine Learning Researcher - MLR	Big Tech	9	L1
Google	Software Engineer, Foundational ML Research	Big Tech	9	L1
Google	Research Engineer, Multi Agent Learning, DeepMind	Big Tech	9	L4
Google	Research Engineer	Big Tech	9	L2
Google	Senior Research Engineer	Big Tech	9	L2
Google	Research Scientist, Efficient AI	Big Tech	9	L3
Google	Research Engineer, Agentic Security & Privacy Post-Training for Gemini, DeepMind	Big Tech	9	L2
Amazon	Member of Technical Staff - Science, Frontier AI & Robotics (FAR)	Big Tech	9	L2
Amazon	Human-Robot Interaction Applied Scientist , Fauna	Big Tech	9	L4
Google	Staff Research Scientist, AI2 Research	Big Tech	9	L1
Apple	AIML - Machine Learning Researcher, MLR	Big Tech	9	L1
Intel	Neuromorphic Applications Researcher- Temporary Position	Semiconductors	9	L6
Wayve	Applied Scientist, Wayve Labs	Robotics	9	L4
Amazon	Member of Technical Staff - Hardware Science, Frontier AI & Robotics (FAR)	Big Tech	9	L6
Amazon	Member of Technical Staff - Science, Frontier AI & Robotics (FAR)	Big Tech	9	L4
Cohere	Machine Learning Intern/Co-op (Fall, 2026)	AI Frontier	9	L2

Frequently asked questions

What is JAX and how common is it in AI job postings?
JAX is a skill in the "Frameworks & Tools" category. It currently appears in 242 active AI roles across 39 companies in our index.
Which companies are hiring for JAX?
The top employers with active AI roles mentioning JAX are: NVIDIA (51), Amazon (38), Google (34), Apple (30), Cohere (15).
Is demand for JAX growing?
Over the last 12 weeks, 138 new AI postings mentioned JAX. Demand is rising — up 139% in the last four weeks compared to the earliest four weeks in the window.
At which stage of AI development is JAX used?
Roles requiring JAX are concentrated in: serving infrastructure (40%), post-training (17%), agents (14%). These stages follow a seven-stage AI lifecycle from data preparation through to shipped product.
What skills are commonly paired with JAX?
Job postings that mention JAX most often also require: PyTorch, Python, Machine Learning, TensorFlow, Computer Architecture.

JAX

Top employers

Often paired with

Demand trend

Frequently asked questions

Demand trend

JAX

Top employers

Often paired with

Frequently asked questions