scimba_torch.neural_nets.structure_preserving_nets

Structure preserving neural networks.

Modules

affine_ode_layers	Affine and constant flow layers for invertible networks.
coupling_layers	Coupling layer for invertible networks.
coupling_symplectic_layers	Symplectic coupling layers for structure-preserving neural networks.
invertible_nn	An invertible neural network made of RealNVP layers.
nilpotent_symplectic_layer	Nilpotent layers for invertible neural networks.
nonlinear_ode_layers	Nonlinear ODE layers for invertible neural networks.
ode_splitted_layer	ODE-based splitted layer for invertible networks.
separated_symplectic_layers	Defines the SympNet class for symplectic neural networks.
split_layer	Split layer for invertible networks.
symplectic_nets	Symplectic neural networks composed of invertible layers.