Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Replace TE patch with TE+dropout patch (#288)
Transformer Engine in Pax currently lacks dropout support. This PR updates the TE patch to add support dropout.
- Loading branch information