Mohammad Amin Nabian
Mohammad Amin Nabian
```[tasklist] ### Tasks - [ ] https://github.com/NVIDIA/modulus/issues/464 - [ ] https://github.com/NVIDIA/modulus/issues/465 - [ ] https://github.com/NVIDIA/modulus/issues/466 - [ ] https://github.com/NVIDIA/modulus/issues/467 - [ ] UNet 3D: Support spatial parallelism ```
### Version Latest ### On which installation method(s) does this occur? _No response_ ### Describe the issue Traceback (most recent call last): File "/code/mnabian/PRs/out/modulus/examples/generative/corrdiff/train.py", line 341, in main training_loop.training_loop( File...
Tracker for the issues related to integrating VFGN ```[tasklist] ### Pytest - [ ] https://github.com/NVIDIA/modulus/issues/451 ```
```[tasklist] ### Integration - [x] Integration of the TPUN - [x] Enable Hydra config - [ ] Add docstrings - [ ] Add license headers - [ ] Add README...
### Is this a new feature, an improvement, or a change to existing functionality? Improvement ### How would you describe the priority of this feature request Critical (currently preventing usage)...
```[tasklist] ### Code structuring - [ ] Split generate.py, train.py and training_loop.py into functions/classes that handle more well-defined tasks - [ ] Reduce redundancy between the above files - [x]...