Optimized tranining recipes for accelerating PyTorch workflows of AI driven surrogates for physical systems
Project description
Modulus Launch (Beta)
Modulus Launch is a PyTorch based deep-learning collection of training recipes and tools for creating physical surrogates. The goal of this repository is to provide a collection of deep learning training examples for different phenomena as starting points for academic and industrial applications. Additional information can be found in the Modulus documentation.
Modulus Packages
Installation
PyPi
The recommended method for installing the latest version of Modulus Launch is using PyPi:
pip install nvidia-modulus.launch
NVCR Container
The recommended Modulus docker image can be pulled from the NVIDIA Container Registry:
docker pull nvcr.io/nvidia/modulus/modulus:23.05
From Source
Package
For a local build of the Modulus Launch Python package from source use:
git clone git@github.com:NVIDIA/modulus-launch.git && cd modulus-launch
pip install --upgrade pip
pip install .
Source Container
To build Modulus Launch docker image:
docker build -t modulus-launch:deploy --target deploy -f Dockerfile .
To build CI image:
docker build -t modulus-launch:ci --target ci -f Dockerfile .
To build any of these images on top of the Modulus base image, you can
--build-arg BASE_CONTAINER=modulus:deploy
to the above commands as shown below:
docker build --build-arg BASE_CONTAINER=modulus:deploy \
-t modulus-launch:deploy --target deploy -f Dockerfile .
Contributing
For guidance on making a contribution to Modulus, see the contributing guidelines
Communication
- Github Discussions: Discuss architectures, implementations, Physics-ML research, etc.
- GitHub Issues: Bug reports, feature requests, install issues, etc.
- Modulus Forum: The Modulus Forum hosts an audience of new to moderate level users and developers for general chat, online discussions, collaboration, etc.
License
Modulus Launch is provided under the Apache License 2.0, please see LICENSE.txt for full license text.
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
Built Distribution
Hashes for nvidia_modulus.launch-0.2.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | a4198f741ea8931cd4feae5953f149e10011378c94982efbca93883bba41d510 |
|
MD5 | a2b21195cc5a889d629d89486c55b463 |
|
BLAKE2b-256 | 03bb917aa62a17f6985593092fce4e942974ac7b4cf6e64a025aad9499185c29 |