patrickvonplaten's picture sanchit-gandhi's picture
Add fp32, flax weights (#5)
1940b90