George Hotz
George Hotz
Rename EXPAND to REPEAT to match einops
It's 961 lines now! Done through the power of refactoring, not code golf.
This still alive?
Would offer a $2,000 bounty for this.
Great! Add some tests to confirm it matches PyTorch GroupNorm and I'll merge
Want to flip it to match PyTorch?
I really don't want it different from PyTorch, I agree ours matches the comments, but people shouldn't have to read the docs if they are migrating, it should just do...
"elementwise_affine on channels" HACK needs to be fixed too, should match the PyTorch behavior in all cases. (raising the atol and rtol is fine btw)
Please match exactly the first 4 args of the PyTorch version. Like why would we name in in_channels when PyTorch names it num_channels? nn is trying to be very high...
Fix stable diffusion in a way PyTorch would like! What is None supposed to do it's been a while since I wrote this