Redirecting to the project codebase:
https://github.com/Project-MONAI/research-contributions/tree/main/lamp-automated-model-parallelism