jax-ml / jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Date Created 2018-10-25 (6 years ago)

Commits 26,855 (last one 3 hours ago)

Stargazers 31,865 (40 this week)

Watchers 328 (0 this week)

Forks 2,971

License apache-2.0

Ranking

RepositoryStats indexes 636,587 repositories, of these jax-ml/jax is ranked #678 (99th percentile) for total stargazers, and #3,354 for total watchers. Github reports the primary language for this repository as Python, for repositories using this language it is ranked #125/130,324.

Other Information

jax-ml/jax has 496 open pull requests on Github, 12,781 pull requests have been merged over the lifetime of the repository.

Github issues are enabled, there are 1,509 open issues and 4,607 closed issues.

There have been 121 releases, the latest one was published on 2025-03-19 (19 days ago) with the name JAX v0.5.3.

Homepage URL: http://jax.readthedocs.io/

All Topics

jax

Star History

Github stargazers over time

Watcher History

Github watchers over time, collection started in '23

Recent Commit History

16,704 commits on the default branch (main) since jan '22

Yearly Commits

Commits to the default branch (main) per year

Issue History

Languages

The primary language is Python but there's also others...

Opengraph Image

updated: 2025-04-08 @ 12:41pm, id: 154739597 / R_kgDOCTkjjQ