SIGN IN SIGN UP

Merge pull request #420 from AI-Hypercomputer:ninatu/fix_heads_per_tile

PiperOrigin-RevId: 933787438
M
maxdiffusion authors committed
4ccaabab2dce40aa4dbb8532b43fba07bbd5562d