Train Models Faster with JAX and MaxText Using NVFP4 on NVIDIA Blackwell | NVIDIA Technical Blog
… To enable it, set the quantization flag in MaxText to switch into the NVFP4 path. Two modes are exposed: quantization=te nvfp4 : NVFP4 with Random Hadamard Transform. Recommended when the convergence under te nvfp4 no rht is not satisfactory. quantization=te nvfp4 no rht : NVFP4 without RHT. …