stfc-sciml / sciml-bench

SciML Benchmarking Suite for AI for Science
MIT License
38 stars 13 forks source link

Parallel version of Hydronet #22

Open juripapay opened 1 year ago

juripapay commented 1 year ago

The current version of Hydronet works only on 1 GPU. The parallel version can be found here: https://github.com/jenna1701/hydronet/tree/schnet_ddp/challenge-1.5 I got the link yesterday, currently testing if the code works.

samueljackson92 commented 1 year ago

The PR for this is #29