# 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. Source: Sumble.

## 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 | Job posts | People | Recent job post |
|---|---:|---:|---|
| [Apple](https://sumble.com/orgs/apple) | 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](https://sumble.com/orgs/google) | 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](https://sumble.com/orgs/nvidia) | 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)](https://sumble.com/orgs/amazon-web-services) | 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](https://sumble.com/orgs/amazon) | 510 | 20 | "Extensive programming skills in Python and PyTorch/JAX" (Human-Robot Interaction Applied Scientist , Fauna, Oct 27, 2026) |

## 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.

| Month | Job posts naming it | All job posts | Per 10,000 job posts |
|---|---:|---:|---:|
| Oct 2025 | 663 | 1,413,673 | 4.7 |
| Nov 2025 | 807 | 1,822,119 | 4.4 |
| Dec 2025 | 537 | 1,236,103 | 4.3 |
| Jan 2026 | 810 | 1,828,002 | 4.4 |
| Feb 2026 | 850 | 1,856,647 | 4.6 |
| Mar 2026 | 713 | 1,850,711 | 3.9 |
| Apr 2026 | 705 | 1,728,886 | 4.1 |
| May 2026 | 728 | 1,821,509 | 4.0 |
| Jun 2026 | 738 | 1,909,496 | 3.9 |
| Jul 2026 | 778 | 1,815,180 | 4.3 |
| Aug 2026 | 801 | 1,835,698 | 4.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.

| Period | Companies naming 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% |

## 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.

| Size | Share | Typical tech page |
|---|---:|---:|
| 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.

| Country | Companies | Share | Typical tech page |
|---|---:|---:|---:|
| United States | 2,262 | 52% | 43% |
| United Kingdom | 318 | 7.3% | 6.4% |
| Germany | 242 | 5.5% | 4.4% |
| India | 195 | 4.5% | 6.9% |
| France | 188 | 4.3% | 4.6% |
| Canada | 143 | 3.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 |

## 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 | Oct 2025 to Aug 2026 | Mar to May 2026 | Jun to Aug 2026 |
|---|---:|---:|---:|---:|
| Machine Learning | 25% | 1,997 | 446 | 472 |
| AI Engineer | 22% | 1,822 | 457 | 529 |
| Software Engineer | 8.9% | 725 | 189 | 235 |
| Researcher | 5.5% | 451 | 92 | 154 |
| Data Scientist | 2.5% | 202 | 55 | 45 |
| Engineer | 2.2% | 176 | 41 | 31 |

## 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.

| Group | Company | Job posts, April to August 2026 |
|---|---|---:|
| AI native | [Sereact](https://sumble.com/orgs/sereact) | 4 |
| AI native | [Arena Physica](https://sumble.com/orgs/arena-ai-com) | 4 |
| AI native | [YTL AI Labs](https://sumble.com/orgs/ytl-ai-labs) | 4 |
| AI native | [Plaud](https://sumble.com/orgs/plaudai) | 2 |
| AI native | [Ivo](https://sumble.com/orgs/latchapp) | 2 |
| Commercial | [LG Electronics Canada](https://sumble.com/orgs/lg-canada-business-solutions) | 8 |
| Commercial | [Byteridge](https://sumble.com/orgs/byteridge-software) | 7 |
| Commercial | [Nomagic](https://sumble.com/orgs/nomagic) | 6 |
| Commercial | [Protezione Civile](https://sumble.com/orgs/meteoschweiz) | 5 |
| Commercial | [Skydropx LATAM](https://sumble.com/orgs/skydropx) | 4 |
| Mid-market | [ARA](https://sumble.com/orgs/applied-research-associates) | 6 |
| Mid-market | [Credit One Bank](https://sumble.com/orgs/credit-one-bank) | 5 |
| Mid-market | [Nscale](https://sumble.com/orgs/nscale-cloud) | 5 |
| Mid-market | [Zycus](https://sumble.com/orgs/zycus-infotech) | 4 |
| Mid-market | [IBS Software](https://sumble.com/orgs/ibs-software-services) | 3 |
| Enterprise | [Intuitive](https://sumble.com/orgs/intuitive-intuitive-surgical) | 5 |
| Enterprise | [Epsilon](https://sumble.com/orgs/epsilon-3) | 4 |
| Enterprise | [Workday](https://sumble.com/orgs/workday) | 3 |
| Enterprise | [Rockwell Automation](https://sumble.com/orgs/rockwell-automation) | 3 |
| Enterprise | [Bundesverwaltung](https://sumble.com/orgs/bundesverwaltung-administration-federale-amministrazione) | 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.

## 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%.

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

Shares are out of the 2,313 companies whose job posts name JAX, not counting firms whose job posts name almost every technology.

## 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.