Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Update Paxml patchlist with TE and config patches for improved perf (#…
…225) - adds Transformer Engine support to Pax - updates GPU configs and default XLA flags for improved performance --------- Co-authored-by: Terry Kong <terryk@nvidia.com>
- Loading branch information