maxtext
maxtext copied to clipboard
Adding FP8 support through NVIDIA Transformer Engine
Opening a draft pull request while we debug some XLA pattern matcher issues.