Best Open Source jax Libraries

A curated list of the most popular GitHub repositories tagged with jax. Select any project to visualize its architecture and dive into the codebase using RepoMind's AI engine.

#1keras-team/keras

Deep Learning for humans

63,864Python

Analyze Code

#2jax-ml/jax

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

34,918Python

Analyze Code

#3d2l-ai/d2l-en

Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.

28,194Python

Analyze Code

#4lukasmasuch/best-of-ml-python

🏆 A ranked list of awesome machine learning Python libraries. Updated weekly.

23,240

Analyze Code

#5ivy-llc/ivy

Convert Machine Learning Code Between Frameworks

14,212Python

Analyze Code

#6arogozhnikov/einops

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

9,398Python

Analyze Code