Here's a google collab notebook built from this. It takes ~2 hours on A100 GPU if you have collab pro. Might work on free account as well.

https://colab.research.google.com/drive/1dklqzK8TDPfbPbyHrk3...