TokenWang's picture WRHC's picture
Convert to BF16 and re-shard (#5)
19bc874