search

NVIDIA / modulus-launch

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

🚀[FEA]: Add nvtx markers to the unified training recipe #88

Open mnabian opened 1 year ago

mnabian 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

Medium

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

To facilitate profiling

Describe any alternatives you have considered

No response