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.interpolation	Interpolation layers
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.share	Sharing modules
inox.nn.ssm	State space model (SSM) layers
inox.nn.state	Stateful modules