Skip to content
Equinox
Module
Initializing search
patrick-kidger/equinox
@PatrickKidger
Equinox
patrick-kidger/equinox
@PatrickKidger
Getting started
All of Equinox
Examples
Examples
Introductory
Introductory
Convolutional Neural Network on MNIST
Train RNN
Advanced
Advanced
Generative score-based diffusion
BERT language model
U-Net implementation
Vision Transformer (ViT)
Image GAN
Features
Features
Freezing parameters
Compatibility with init-apply libraries
Stateful operations (e.g. BatchNorm)
Autoparallelism (e.g. multi-GPU)
Serialisation (with hyperparameters)
Awesome list
Basic API
Basic API
Modules
Modules
Module
Module
Table of contents
Module
Extra features
Neural network layers
Neural network layers
Linear
Convolutional
Recurrent
Attention
Activations
Pooling
Dropout
Normalisation
Embeddings
Multi-layer perceptron
Sequential
Training/Inference
Sharing layers
Stateful operations
Transformations
Manipulating PyTrees
Advanced API
Advanced API
Clear caches
Debugging tools
Enumerations
Runtime errors
Pretty printing
Serialisation
Misc
Misc
FAQ
Tricks (ensembles, surgery, custom initialisations, ...)
Citation
Module
¤
equinox.Module
¤
Module()