google-deepmind/sonnet

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

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Models and examples built with TensorFlow

Library of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.

An Open Source Machine Learning Framework for Everyone

Tensor library for machine learning

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

You like pytorch? You like micrograd? You love tinygrad! ❤️

CoreNet: A library for training deep neural networks