Discofork
Repository brief

jax-ml/jax

Read the upstream summary on the left, browse the cached forks below it, and load each fork comparison into the right-hand panel.

Cached analysis
cached 2026-03-30T15:46:16.621Z
1mo ago

jax-ml/jax

jax-ml/jax is a large, active Python numerical computing library for composable transformations of Python+NumPy code, including automatic differentiation, vectorization, and JIT compilation to accelerators. It is very widely used and actively developed, with 35k+ stars, 3.4k+ forks, and recent commits on March 30, 2026.

GitHub
Loading tags...
Stars35,258
Forks3,493
Default branchmain
Last pushed2026-03-30T15:38:45Z
Recommended shortcuts

Jump straight into Discofork's strongest cached fork picks, or open a compare view in one click.

Forks

Choose a fork to inspect

10 of 10 fork briefs
Selected

Choose this fork if ROCm is the target platform and you need the AMD-specific runtime, packaging, and CI/test work. Choose upstream if you value the newest JAX features and broader accelerator parity more than ROCm specialization.

Choose this fork only if you need an old, frozen JAX baseline or existing fork-specific edits. For new work, current upstream is the better choice because this fork is far behind and likely missing many modern features and fixes.

Prefer upstream unless you specifically need this fork’s historical experimental features or older semantics. This fork looks like a stale, highly divergent snapshot with a few niche additions, not a maintained alternative platform.

Prefer this fork only if you need its TPU/GPU internals or experimental accelerator features. For general JAX use, upstream is a better default because this fork is materially behind and likely to lag on fixes, compatibility, and release-quality maintenance.

Prefer this fork only if you must keep an older JAX API surface alive. For new work, or anything expecting modern JAX features and active maintenance, upstream is the clear choice.

Choose this fork only if you need its exact experimental JAX behavior or historical accelerator patches. For most users, upstream JAX is the safer choice because this fork is far behind and appears to have lost or disabled substantial upstream functionality.

Prefer upstream unless you specifically need this fork's older experimental backend work or are continuing an existing dependency on its 2023-era codebase. For most adopters, the staleness and divergence are the main reasons to avoid it.

Prefer upstream JAX unless you specifically want this fork’s experimental Numba-oriented direction or are studying old internals. For production or current ML/accelerator work, the fork is too stale and diverged to be a safe default.

Prefer upstream unless you specifically need an old JAX snapshot; this fork is too stale to adopt for modern accelerator, compiler, or library work.

Prefer the upstream project unless you specifically need legacy JAX behavior or historical reproducibility. This fork is so far behind and so stale that it is not a good adoption target for modern accelerator or plugin work.