search

NVIDIA / modulus-launch

Repo of optimized training recipes for accelerating PyTorch workflows of AI driven surrogates for physical systems
Apache License 2.0
54 stars 27 forks source link

🚀[FEA]: Enable CUDA Graphs and AMP for DLWP example #93

Closed ktangsali closed 1 year ago

ktangsali commented 1 year ago

Is this a new feature, an improvement, or a change to existing functionality?

New Feature

How would you describe the priority of this feature request

Low (would be nice)

Please provide a clear description of problem you would like to solve.

The DLWP example can be refactored to enable CUDA Graphs and AMP support

Describe any alternatives you have considered

No response