discover · compare · collections · requirements · contribute · feed · watching · stats · help with your tokens
google/flax
Flax is a neural network library for JAX that is designed for flexibility.
History and source changes · Reports and owner responses · Compare
Similar
- jax-ml/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Why this match
topics jax; language differs: Jupyter Notebook → Python
- unifyai/ivy
Convert Machine Learning Code Between Frameworks
Why this match
topics jax; language differs: Jupyter Notebook → Python
- keras-team/keras
Deep Learning for humans
Why this match
topics jax; language differs: Jupyter Notebook → Python
- google/trax
Trax — Deep Learning with Clear Code and Speed
Why this match
topics jax; language differs: Jupyter Notebook → Python
- arogozhnikov/einops
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
Why this match
topics jax; language differs: Jupyter Notebook → Python; license differs: Apache-2.0 → MIT
- probml/pyprobml
Python code for "Probabilistic Machine learning" book by Kevin Murphy
Why this match
topics jax; license differs: Apache-2.0 → MIT
- HIPS/autograd
Efficiently computes derivatives of NumPy code.
Why this match
topics jax; language differs: Jupyter Notebook → Python; license differs: Apache-2.0 → MIT
- d2l-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.
Why this match
topics jax; language differs: Jupyter Notebook → Python