Try for free
Try for free

Companies using JAX

4,681 companies in our data use JAX, and 3,139 of them name it in job posts they wrote. Hiring for it is steady, with 8,130 job posts naming it from October 2025 to August 2026. Machine Learning roles mention it most.

Data as of November 9, 2026 · Markdown version

Which companies name JAX in the most job posts?

Apple, Google and NVIDIA are among the companies with the most job posts naming JAX. Amazon Web Services (AWS) is the newest of the five companies in the table, with its first job post naming it in December 2023.
Company Posts People Recent job post
Apple Scientific and Technical Services · first seen Jul 2022 2,134 34 "Industry experience of coding in Python and PyTorch or JAX"Machine Learning Researcher - Apple Services Engineering - Generative Audio, Nov 7, 2026
Google Scientific and Technical Services · first seen Aug 2022 1,289 142 "…it's easy to take your existing ML workloads on leading ML frameworks (PyTorch and JAX) and move them to the cloud to take advantage of this great capability."Senior Staff Software Engineer, Machine Learning, ML Training, Nov 7, 2026
NVIDIA Scientific and Technical Services · first seen Aug 2022 1,120 58 "Design and implement program transformations on JAX programs (jaxpr) for large-scale training, on top of unmodified JAX and XLA: pipeline schedules, placement…"Senior Compiler Engineer - AI, Nov 7, 2026
Amazon Web Services (AWS) Information · first seen Dec 2023 791 17 "They should possess a working knowledge of Kubernetes and associated machine learning frameworks such as PyTorch, JAX, Kubeflow, Kueue, etc."Senior Software Development Engineer, AWS Kubernetes (K8s), Nov 4, 2026
Amazon Retail Trade · first seen Jul 2022 510 20 "Extensive programming skills in Python and PyTorch/JAX"Human-Robot Interaction Applied Scientist , Fauna, Oct 27, 2026
Showing 1 to 5 of 3,139 companies with job posts Sign up to see all companies using JAX ›

What else do we know about JAX?

Is hiring for JAX growing?

Hiring for JAX barely changed, at 4.2 in every 10,000 job posts from June to August 2026 against 4.0 from March to May. From October to December 2025 it was 4.5.
Share of job posts naming JAX by month, October 2025 to August 2026, in job posts in every 10,000 Moves from 4.7 in October 2025 to 4.4 in August 2026, in job posts in every 10,000. 5.4 4.0 2.7 1.3 0 4.4 Oct '25 Nov Dec Jan '26 Feb Mar Apr May Jun Jul Aug
Show the data
MonthJob posts naming JAXAll job postsPer 10,000
Oct 20256631,413,6734.7
Nov 20258071,822,1194.4
Dec 20255371,236,1034.3
Jan 20268101,828,0024.4
Feb 20268501,856,6474.6
Mar 20267131,850,7113.9
Apr 20267051,728,8864.1
May 20267281,821,5094.0
Jun 20267381,909,4963.9
Jul 20267781,815,1804.3
Aug 20268011,835,6984.4

How many companies named JAX in a job post for the first time in recent months?

314 companies named JAX in a job post for the first time from June to August 2026, against 254 from March to May. From October 2025 to August 2026, 1,001 companies named it in a job post for the first time.
March to May 2026 254
June to August 2026 314
October 2025 to August 2026 1,001

Companies whose first job post naming it appeared in that period. Recruiters, staffing firms and job boards are left out.

How do we know which companies use JAX?

3,139 of the 4,681 companies name JAX in their own job posts. 445 show up in both job posts and employees' profiles.
How we know Companies Share
Job posts name JAX
3,139 67%
Employees' profiles list JAX
1,542 33%

4,681 companies. Each is counted once, under the strongest evidence we have for it.

What size are the companies that use JAX?

Companies with fewer than 50 employees are 1.5 times as common among JAX users as on the typical tech page. 833 of its companies have more than 5,000 employees.

Share of its companies Typical tech page

ShareTypical
Under 50 employees
32% 22%
50 to 999
36% 38%
1,000 to 5,000
15% 17%
Over 5,000
18% 23%

Where are JAX companies based?

More than half of JAX companies are in the United States (52%). India is less common than usual, at 4.5% against 6.9% on the typical tech page.
ShareTypical
United States
52% 43%
United Kingdom
7.3% 6.4%
Germany
5.5% 4.4%
India
4.5% 6.9%
France
4.3% 4.6%
Canada
3.3% 3.6%

Companies in 82 countries and territories. Shares are of companies with a known headquarters country.

Show the data
CountryCompaniesShareTypical tech page
United States2,26252%43%
United Kingdom3187.3%6.4%
Germany2425.5%4.4%
India1954.5%6.9%
France1884.3%4.6%
Canada1433.3%3.6%

Which teams work with JAX?

Job posts most often place JAX with AI, machine learning and data science teams. Next are R&D and research teams, at 34 and 33 companies.
Team Companies
AI
65
Machine learning
49
Data science
38
R&D
34
Research
33
ML
31

What else do JAX companies use?

The tool most tied to JAX is DeepSpeed. 11% of JAX companies use it, 62 times the usual rate for their size. SGLang comes second, 56 times as common as usual, at 6.9%.

Share of the 2,313 JAX companies counted here whose job posts also name the tool Typical for companies the same size

Tool Also name it Typical Times as common
DeepSpeed 255 companies
11% 0.18% 62x
SGLang 159 companies
6.9% 0.12% 56x
Ray 430 companies
19% 0.47% 39x
NVIDIA Triton 406 companies
18% 0.47% 37x
Tool Also name it Typical Times as common
TensorRT 385 companies
17% 0.49% 34x
vLLM 395 companies
17% 0.54% 31x
ONNX 331 companies
14% 0.48% 30x
Weights & Biases 275 companies
12% 0.41% 29x

Which companies started naming JAX in job posts recently?

259 companies with 50 or more employees picked up JAX in their job posts from April to August 2026. Another 33 were AI-native companies, which we list separately.
AI native any size · 33 new
Company Posts
Sereact 4
Arena Physica 4
YTL AI Labs 4
Plaud 2
Ivo 2
Commercial 50 to 999 employees · 119 new
Company Posts
LG Electronics Canada 8
Byteridge 7
Nomagic 6
Protezione Civile 5
Skydropx LATAM 4
Mid-market 1,000 to 5,000 employees · 70 new
Company Posts
ARA 6
Credit One Bank 5
Nscale 5
Zycus 4
IBS Software 3
Enterprise over 5,000 employees · 70 new
Company Posts
Intuitive 5
Epsilon 4
Workday 3
Rockwell Automation 3
Bundesverwaltung 3

Companies that named JAX in a job post for the first time between April and August 2026, ranked by their job posts in those months. AI-native companies get a group of their own, whatever their size. Recruiters and job boards are left out.

Which roles mention JAX most?

Researcher job posts naming JAX grew the fastest among its top roles: 154 from June to August 2026, up from 92 from March to May. Machine Learning posts naming it numbered 446 from March to May 2026 and 472 from June to August.
Role Share Posts
Machine Learning
25% 1,997
AI Engineer
22% 1,822
Software Engineer
8.9% 725
Researcher
5.5% 451
Data Scientist
2.5% 202
Engineer
2.2% 176
Show the data
RoleShareOct 2025 to Aug 2026Mar to May 2026Jun to Aug 2026
Machine Learning25%1,997446472
AI Engineer22%1,822457529
Software Engineer8.9%725189235
Researcher5.5%45192154
Data Scientist2.5%2025545
Engineer2.2%1764131

Which project themes come up in JAX job posts?

Teams most often pair JAX with Generative AI (2,023 job posts), then broad digital transformation, then Cloud Cost Optimization and Microservices Migration.
Project theme Job posts
Generative AI 2,023
Digital Transformation 1,121
Cloud Cost Optimization 209
Microservices Migration 70
About this data

Sumble built this page. Sumble gives sales, marketing and other GTM teams technographics and hiring signals for the companies they sell to: the technologies each company uses, which teams are hiring and for what, and who reports to whom. That helps a team know who to contact, why now, and what to say.

Sumble covers more than 3.6 million companies, 93 million job posts collected since 2022 and 194 million employee profiles, and tracks more than 37,000 technologies. It finds which technologies a company uses by reading its job posts and its employees' public profiles.

Sumble has a free plan, and the same data is available through its API and MCP server.

A company counts when its own job posts or its employees' public profiles name the technology. Hiring is that technology's share of all job posts collected each month. Quotes are copied word for word from the companies' job posts.

Every figure on this page comes from Sumble's own analysis, and you won't find it outside Sumble.