Hi all, great work!

Was trying to reproduce your example and two small typos came to view.

1. the minimal torch requirement should be 1.11.0 which is the first version that introduces antialias argument to nn.functional.interpolate needed by GEM/gem/gem_utils

2. in the demo examples, gem_model.visualize should actually be gem.visualize, right?

Happy to submit a PR for this if preferred but those are very typos.

Again, great work!
Cheers!