SoftwareFramework

JAX

Project record

Maintained by JAX core team (Google open-source project)3 4 5

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 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 list

Public materials checklist

Items for a framework under USASI rubric v0.1. 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)

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

Report a correction Edit this entry

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