I’ve tried to use @vrandezo’s Python notebook from https://github.com/vrandezo/colabs to compare/recreate results, but ran into:

OSError: [Errno 12] Cannot allocate memory