Float8E5m2fnuzΒΆ FP8 type E5M2FNUZ. PyTorchJAX (Core)NumPyFlax NNX PyTorchAPI: torch.float8_e5m2fnuzStrategy: Direct MappingOfficial Docs βJAX (Core)API: jnp.float8_e5m2fnuzStrategy: Direct MappingOfficial Docs βNumPyAPI: βStrategy: Custom / PartialFlax NNXAPI: jnp.float8_e5m2fnuzStrategy: Direct MappingOfficial Docs β