search

kmzzhang / nbi

Package for Neural Posterior Estimation and Importance Sampling focused on Astronomical Applications
Other
35 stars 4 forks source link

Support for multi-node inference #3

Open kmzzhang opened 1 year ago

kmzzhang commented 1 year ago

Rewrite in DistributedDataParallel and test on Perlmutter