This is cool. I want to give it a try--I hope my MacBook supports CUDA!
As an alternative, feel free to play around with it and get familiar with the setup process in the Google colab notebook.
https://colab.research.google.com/github/hardik0/AI-basketba...
Unfortunately, I don't have written a notebook for it. Maybe you could try to figure it out through the source code? The code itself is actually prettly simple and short.