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

2 years 6 months

Backers

Google (Creator and maintainer)

Size score (1 to 10, higher is better)

5.5

Trend score (1 to 10, higher is better)

9.0

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 May 31, 2021.

Statistic

Lifetime

Last 12 Months

Commits

18,091

13,614

Lines committed

3,389,700

2,832,105

Unique committers

330

178

Core committers

10

8

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

Similar Projects

Project

Size Score

Trend Score

Byline

Flashlight

5.5

7.5

A C++ standalone library for machine learning

Modin

4.5

6.25

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

NumPy

8.75

6.5

The fundamental package for scientific computing with Python.

Pandas

7.5

7.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.5

6.75

GPU dataframe library