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.
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
Scikeras : Give Scikit-Learn like API to your Keras Neural Networks
Skorch: Give Scikit-Learn like API to your PyTorch Neural Networks