Train Models Faster with JAX and MaxText Using NVFP4 on NVIDIA Blackwell
NVIDIA introduces NVFP4 precision support in JAX and MaxText to accelerate large language model pretraining on Blackwell GPUs, improving throughput and reducing training time and costs.