JAX-based neural network library
A cross-platform Python library for differentiable programming
Flax is a neural network library for JAX
TensorFlow-based neural network library
Project for processing neural networks and rendering to gain insights
A library for easy and efficient manipulation of tensor networks
Neural network library for developers