Multi GPU Code

[pk1996]

Hi

Thanks for this work.

The README says only single GPU is supported. But your code has DDP code. Is it fully supported yet?

Any updates on this end?

Thanks