eqy

@pytorchmergebot rebase

@pytorchmergebot rebase

@pytorchmergebot merge

Thanks @antferdom --- we're starting to look into this now and are currently working on verifying if the performance looks promising compared to the current native PyTorch implementation(s) CC @Aidyn-A...

@Aidyn-A and @mikaylagawarecki are currently working on it