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 Datapipe benchmarks that clearly show/compare performance #117

Open loliverhennigh opened 1 year ago

loliverhennigh 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.

Implement datapipe benchmarks as an example that is easy for users to run. This will both allow us to show new users a compelling reason to use modulus while also allowing us to iterate on datapipe ideas such as zarr stores.

Describe any alternatives you have considered

No response