JAX

JAX is Autograd and XLA, brought together for high-performance machine learning research.

Logo

../_images/google_jax-small.png

Website

N/A

Repository

https://github.com/google/jax

Byline

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

License

Apache 2.0

Project age

4 years

Backers

Google (Creator and maintainer)

Size score (1 to 10, higher is better)

6.25

Trend score (1 to 10, higher is better)

6.5

Education Resources

URL

Resource Type

Description

https://jax.readthedocs.io/en/latest/notebooks/quickstart.html

Documentation

Official project documentation.

Git Commit Statistics

Statistics computed using Git data through November 30, 2022.

Statistic

Lifetime

Last 12 Months

Commits

61,114

21,111

Lines committed

8,218,891

2,045,886

Unique committers

521

204

Core committers

13

14

../_images/google_jax-monthly-commits.png

Similar Projects

Project

Size Score

Trend Score

Byline

Flashlight

7.5

6.5

A C++ standalone library for machine learning

Modin

5.0

7.5

Speed up your Pandas workflows by changing a single line of code

Pandas

7.25

5.25

Flexible and powerful data analysis / manipulation library for Python, providing labeled data structures similar to R data.frame objects, statistical functions, and much more.

cuDF

7.0

6.0

GPU dataframe library

statsmodels

7.25

5.75

Statistical modeling and econometrics in Python