search

noahgolmant / pytorch-hessian-eigenthings

Efficient PyTorch Hessian eigendecomposition tools!
MIT License
360 stars 43 forks source link

Add DistributedDataParallel and MPI Integration #32

Open noahgolmant opened 3 years ago

noahgolmant commented 3 years ago

Add and test multi-GPU support and MPI integration. I need GPUs for this.