inox.nn

Neural networks, layers and modules

Submodules

inox.nn.activation	Activation functions
inox.nn.attention	Attention layers
inox.nn.container	Container modules
inox.nn.dropout	Dropout layers
inox.nn.einops	Einstein-like tensor operations
inox.nn.linear	Linear layers
inox.nn.module	Base modules
inox.nn.normalization	Normalization layers
inox.nn.padding	Padding layers
inox.nn.pooling	Pooling layers
inox.nn.recurrent	Recurrent layers
inox.nn.ssm	State space model (SSM) layers