Plans for more puzzles?

[kaiavintr]

I nominate amin (or amax), nonzero, and, sort.

[srush]

I'm always happy for a PR if you are interested.

I think these ones are quite difficult to do though. Would be curious how you pull it off. Nonzero is a particularly nasty one because it has dynamic output size (but maybe you could do it with padding)

Maybe check out autodiff-puzzles for some sorting fun.

[kaiavintr]

I don't want to give spoilers, but nonzero was easy once I realized the trick (and it's an important trick, so I think it would make a good addition). Sort was definitely challenging, and took many tries to get it under 80 characters though.

I am definitely going to check out your other puzzles! Thank you so much for making these!

[dantp-ai]

torch.gather would be a nice addition to this awesome collection of puzzles.