bthens/model-collection

This project contains a collection of deep learning models developed by the AI4Sim team with various partners. This is is structured on the basis of use-cases providing canonical PyTorch Lightning pipelines allowing to train neural network models that are able to surrogate various physical processes.

PythonApache-2.0