Non-distributed multi-GPU training

lab-cosmo / metatrain

Training and evaluating machine learning models for atomistic systems.

https://lab-cosmo.github.io/metatrain/

BSD 3-Clause "New" or "Revised" License

13 stars 3 forks source link

Open frostedoyster opened 3 weeks ago

frostedoyster commented 3 weeks ago

Following the discussion in #239, it would be nice to have multi-GPU training in non-distributed environments