Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Tags: alxndrTL/apex

Tags

25.04

Toggle 25.04's commit message

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature.
Graph Compiler For Multi-stream (NVIDIA#1886)

* Graph compiler for multi-stream.

Co-authored-by: Wil Kong <[email protected]>
Co-authored-by: Rundong Li <[email protected]>

* Reuse streams across torch compile graphs.

---------

Co-authored-by: Rundong Li <[email protected]>

24.04.01

Toggle 24.04.01's commit message

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature.
Enhance Distributed Fused Adam (NVIDIA#1794)

23.08

Toggle 23.08's commit message
source of apex in nvcr.io/nvidia/pytorch:23.08-py3

23.07

Toggle 23.07's commit message
source of apex in nvcr.io/nvidia/pytorch:23.07-py3

23.06

Toggle 23.06's commit message
source of apex in nvcr.io/nvidia/pytorch:23.06-py3

23.05

Toggle 23.05's commit message
source of apex in nvcr.io/nvidia/pytorch:23.05-py3

22.03

Toggle 22.03's commit message

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
[transformer] Update `build_model` function to support encoder&decode…

…r model (NVIDIA#1307)

* update build_model to support enc&dec model

* fix typo: cur_sargs -> cur_args

* enc&dec path: correctly update pre/post process