models

Submodule containing the neural network models.

Contains the main LightningModule in mldft.ml.models.mldft_module and the components for the neural networks in mldft.ml.models.components.

Modules

components	Components for the ML models.
mldft_module	Lightning module for the neural network.