SoftwareFramework

JAX

JAX is a framework maintained by JAX core team (Google open-source project)3 4 5. Its source code is public1 3 under the Apache License 2.02.

JAX is a Python library for array computation and program transformation, including automatic differentiation, just-in-time compilation, and vectorization of NumPy-style code. It compiles programs with the XLA compiler to run on CPUs, GPUs, TPUs, and other accelerators.1 5

Repository: GitHub repository (external site: github.com)
Documentation: Documentation (external site: docs.jax.dev)
License: License (external site: github.com)

Last reviewedSep 29, 2026Entry updated Sep 29, 2026Documented release Unknown

Availability and license

Overall availability

Public

Documented as available to the general public. Access conditions and license terms may still apply.1 2

Availability fact review: not separately recorded

Availability is separate from permission: read the license before using or redistributing.

Apache License 2.0 (external site: github.com)2
Applies to: Code · SPDX Apache-2.0 · on the rubric’s OSI-approved software-license list · fact-level rights review not recorded
Plain-language guide to the Apache License 2.0

Component reuse rights

A readable or downloadable component is not automatically reusable. These indicators concern recorded license evidence, not system certification.

weights: Unknown — no complete fact-level rights review
code: Unknown — no complete fact-level rights review
data: Unknown — no complete fact-level rights review
documentation: Unknown — no complete fact-level rights review

No complete system-rights review is recorded for this release.

Public materials checklist

Items for a framework under USASI rubric v0.2. Unknown means unassessed or insufficient evidence.

Public materials checklist for JAX
Item	Status	Notes and evidence
Source codeIs the source code publicly readable?	Public	Developed in the open on GitHub at jax-ml/jax.1 3
DocumentationIs user documentation published?	Public	User and developer documentation is published at docs.jax.dev.3 4
InstallationAre installation instructions or packages publicly available?	Public	The README gives pip install commands for CPU, NVIDIA GPU, Google TPU, and AMD GPU builds.1 5
Supported platformsAre supported operating systems or hardware documented?	Public	The README tabulates support by operating system and accelerator, covering Linux, macOS, and Windows with CPU, NVIDIA GPU, Google TPU, AMD GPU, Apple GPU, and Intel GPU at varying support levels, some marked experimental.1
Release statusAre versioned releases published?	Public	Versioned releases are published on PyPI; version 0.11.2 was released on September 17, 2026.5

What it is useful for

The project describes JAX as designed for high-performance numerical computing and large-scale machine learning, and as a base for domain-specific libraries built on top of it.1 3

Organization context

Google (Alphabet)

In the news

Dated, sourced updates in this catalog's news that mention this entry.

Oct 19, 2026 · Release
JAX 0.12.0 adds a proxy backend and removes jax.custom_remat
The JAX project published JAX v0.12.0 on GitHub on October 19, 2026. The release notes list new features including jax.lax.polynomial for polynomial evaluation, jax.custom_vjp.defremat for customizing rematerialization, a registered proxy backend that connects to an IFRT Proxy server such as Pathways on Cloud, jax.sharding.no_collectives, and support for DLPack 1.0. Breaking changes listed include the removal of jax.custom_remat in favor of jax.custom_vjp.defremat, a requirement on x86-64 for an Intel Haswell or newer CPU with AVX2 and FMA support, and a minimum of NumPy 2.2. USASI has not tested this release.
- Google (Alphabet)
- JAX

U.S. eligibility

Project eligibility rests on documented governing or maintaining entities, not on contributors.

Eligible · basis: U.S.-governed project

JAX's documentation says the project is led by the JAX core team, that it follows Google's Open Source Community Guidelines, and that contributions require Google's Contributor License Agreement; the PyPI package lists the JAX team with a google.com contact address. The README calls JAX a research project rather than an official Google product. Google's parent Alphabet Inc. has its principal executive offices in Mountain View, California, per its fiscal 2025 Form 10-K.3 4 5 1 6

Assessed Sep 29, 2026

Sources

1.
jax-ml/jax (GitHub) (external site: github.com)
JAX project · Repository · accessed Sep 29, 2026
2.
JAX LICENSE (Apache License 2.0) (external site: raw.githubusercontent.com)
JAX project · License · accessed Sep 29, 2026
3.
About the project — JAX documentation (external site: docs.jax.dev)
JAX project · Documentation · accessed Sep 29, 2026
4.
Contributing to JAX — JAX documentation (external site: docs.jax.dev)
JAX project · Documentation · accessed Sep 29, 2026
5.
jax on PyPI (external site: pypi.org)
Python Package Index · Release notes · accessed Sep 29, 2026
6.
Alphabet Inc. Form 10-K for the fiscal year ended December 31, 2025 (external site: sec.gov)
Alphabet Inc. (via U.S. Securities and Exchange Commission EDGAR) · Filing · published Feb 5, 2026 · accessed Sep 29, 2026

Email a correction Report a correction Edit this entry

This listing is not an endorsement, a safety assessment, or a federal approval.

Cite this page

APA style

USASI. (2026, September 29). JAX. United States of America Superintelligence. https://unitedstatesofamericasuperintelligence.com/open/jax/

BibTeX

@misc{usasi_jax,
  author = {{USASI}},
  title = {JAX},
  year = {2026},
  month = sep,
  howpublished = {\url{https://unitedstatesofamericasuperintelligence.com/open/jax/}},
  note = {United States of America Superintelligence. Last updated 2026-09-29}
}

The date is when this page was last updated. Add the date you read it if your style needs one.

For maintainers: a “Listed in USASI” badge

Links to this page. It says the entry is listed, not endorsed or rated. About the badge

Markdown (for a README)

[![Listed in the USASI catalog](https://unitedstatesofamericasuperintelligence.com/badges/listed.svg)](https://unitedstatesofamericasuperintelligence.com/open/jax/)

HTML

<a href="https://unitedstatesofamericasuperintelligence.com/open/jax/"><img src="https://unitedstatesofamericasuperintelligence.com/badges/listed.svg" alt="Listed in the USASI catalog" height="20"></a>

Availability and license

Component reuse rights

Public materials checklist

What it is useful for

Organization context

In the news

JAX 0.12.0 adds a proxy backend and removes jax.custom_remat

Explore related information

U.S. eligibility

Sources

Share or cite this page

Support Us