bestofgithub.dev

← Back to the directory

google/flax

View on GitHub

Flax is a neural network library for JAX that is designed for flexibility.

google/flax is #8,713 on the bestofgithub Elo ladder, in Bronze V, with a rating of 1,365 (−13 this week). It has 7,366 stars.

Bronze V

1,365 Elo

  • #8,713 on the ladder
  • +9 today · −13 this week
  • Peak 1,393
10 Elo to Bronze IV

Stars

7,366

  • +1 today
  • +0 this week
  • – this month

Repository

Forks
850
Language
Jupyter Notebook
Last push

Achievements

bestofgithub badgeMaintainer? Show your league in your README

Card: your league's emblem, from Copper to Legendary

[![bestofgithub](https://bestofgithub.dev/badge/google/flax.svg?style=card)](https://bestofgithub.dev/repo/google/flax)

Flat: fits beside your other badges

[![bestofgithub](https://bestofgithub.dev/badge/google/flax.svg)](https://bestofgithub.dev/repo/google/flax)

Install

git
git clone https://github.com/google/flax.git

Similar repositories

jax-ml/jax

Open on GitHubPython

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

  • 1,805−5 Elo, change today
  • 36.6k stars
  • 4k forks
  • 328 watchers
git
git clone https://github.com/jax-ml/jax.git
jax-v0.12.0.124d ago
  • jax

unifyai/ivy

Open on GitHubPython

Convert Machine Learning Code Between Frameworks

  • 1,516−7 Elo, change today
  • 14.1k stars
  • 5.4k forks
git
git clone https://github.com/unifyai/ivy.git
  • jax
  • numpy
  • python
  • pytorch

google/trax

Open on GitHubPython

Trax — Deep Learning with Clear Code and Speed

  • 1,371+1 Elo, change today
  • 8.3k stars
  • 814 forks
  • archived
git
git clone https://github.com/google/trax.git
  • deep-learning
  • deep-reinforcement-learning
  • jax
  • machine-learning

arogozhnikov/einops

Open on GitHubPython

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

  • 1,380−1 Elo, change today
  • 9.6k stars
  • 409 forks
git
git clone https://github.com/arogozhnikov/einops.git
  • cupy
  • deep-learning
  • einops
  • jax

lukasmasuch/best-of-ml-python

Open on GitHub

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

  • 1,628−10 Elo, change today
  • 23.8k stars
  • 3.1k forks
git
git clone https://github.com/lukasmasuch/best-of-ml-python.git
  • automl
  • chatgpt
  • data-analysis
  • data-science

README

Loading README…