✔️ Linting Passed

All linting checks passed. Your pull request is in excellent shape! ☀️

_{Generated for commit: eb8836c. Link to the linter CI: here}

Thank you @ogrisel for the quick reply!

Do you have a chance to run the tests on a CUDA host or do you want somebody else to do it? (I can do it).

Yes, I tested on CUDA in Google Colab and on my Mac (using mps). All green on my side ✅

Thank you very much!:)

@EdAbati Just for some guidance, how did you run this code in google colab so that we can test with gpus?

Thanks again @OmarManzoor :)

I just published the Google Colab notebook in a gist: https://gist.github.com/EdAbati/ff3bdc06bafeb92452b3740686cc8d7c

Thanks again @OmarManzoor :)

I just published the Google Colab notebook in a gist: https://gist.github.com/EdAbati/ff3bdc06bafeb92452b3740686cc8d7c

Thank you for sharing!