Flax is a flexible high-level framework designed on top of JAX to make the process of creating neural networks easier. Tutorials cover a guide on how we can create neural networks using Flax.
Sonnet is a deep learning framework developed on the top of Tensorflow by deepmind. It let us design deep neural networks using its easy-to-use API.
Tutorial guides individual on how to use Haiku to create simple neural networks. Haiku is built on top of JAX and is created to make life of ML developer easier.
Tutorial covers in details how we can use high-level API of JAX library available through 'stax', 'nn' and 'optimizers' modules to create Neural Networks.
Create Simple PyTorch Neural Networks using 'torch.nn' Module
Guide to Create Simple Neural Networks using PyTorch
Guide to Create Simple Neural Networks using JAX
Scikeras : Give Scikit-Learn like API to your Keras Neural Networks
Skorch: Give Scikit-Learn like API to your PyTorch Neural Networks
Neural Network Transfer Learning using Pytorch