[sualehasif]

If you know of any other cool examples or would like us to try something out and add to the repo, please let us know.

If you have access, the repo should be fairly replicable.

[Iamnotthatbad]

Your example of the commenting doesn't seem like "well-commented code" to me.

Just re-stating what is in code in plain language isn't helpful and is kind of annoying to read.

Can you explain why you think it is helpful?

[sualehasif]

The commenting was for the model to learn what the types were so it could translate it to gather. Doing that one shot is surprising to me because it is somewhat hard to get pytorch tensors to work out to the correct shapes.