PyTorch Equivalent?

[Leo-T-Zang]

Hi @smsharma

Thanks for creating these amazing notebooks! I wonder if you know/have PyTorch Equivalent repo.

Thanks!