- 13 Mar, 2025 2 commits
- 20 Jun, 2023 1 commit
-
-
Pruthvi Madugundu authored
- Cherry-pick of https://github.com/NVIDIA/apex/pull/1669
-
- 12 Jun, 2023 1 commit
-
-
flyingdown authored
2.添加环境变量APEX_ROCBLAS_GEMM_ALLOW_HALF用于控制是否使用fp16r 3.添加dcu版本信息 whl包名修改 readme更新安装步骤
-
- 07 Jul, 2022 1 commit
-
-
Masaki Kozuki authored
* remove pyprof * remove reparameterization * remove pyprof test * clean up
-
- 18 Mar, 2022 1 commit
-
-
eqy authored
* update ngc link and dockerhub container tag * update * update * update * Update README.md Co-authored-by:Masaki Kozuki <mkozuki@nvidia.com>
-
- 08 Mar, 2022 2 commits
-
-
Masaki Kozuki authored
This reverts commit 74e04667.
-
Masaki Kozuki authored
-
- 09 Dec, 2021 1 commit
-
-
Hubert Lu authored
-
- 29 Oct, 2021 1 commit
-
-
Peng authored
-
- 10 Feb, 2021 1 commit
-
-
Shoufa Chen authored
* copy-paste friendly * fix import container_abcs issue Nightly PyTorch has removed `container_abcs` from `torch._six`. https://github.com/pytorch/pytorch/commit/58eb23378f2a376565a66ac32c93a316c45b6131#diff-b3c160475f0fbe8ad50310f92d3534172ba98203387a962b7dc8f4a23b15cf4dL35 * fix import container_abcs issue Nightly PyTorch has removed `container_abcs` from `torch._six`. https://github.com/pytorch/pytorch/commit/58eb23378f2a376565a66ac32c93a316c45b6131#diff-b3c160475f0fbe8ad50310f92d3534172ba98203387a962b7dc8f4a23b15cf4dL35 * keep existing for pytorch1.7 and earlier
-
- 16 Dec, 2020 1 commit
-
-
lcskrishna authored
-
- 15 Dec, 2020 1 commit
-
-
lcskrishna authored
-
- 04 Dec, 2020 1 commit
-
-
Stas Bekman authored
-
- 21 Aug, 2020 1 commit
-
-
Chaitanya Sri Krishna Lolla authored
-
- 01 Jun, 2020 1 commit
-
-
mcarilli authored
Co-authored-by:Michael Carilli <mcarilli@nvidia.com>
-
- 29 May, 2020 2 commits
-
-
Chaitanya Sri Krishna Lolla authored
-
lcskrishna authored
-
- 10 Oct, 2019 1 commit
-
-
mcarilli authored
-
- 08 Oct, 2019 1 commit
-
-
Jan Schlüter authored
-
- 27 Aug, 2019 1 commit
-
-
ptrblck authored
* add state_dict, load_state_dict * add test_restoring, test_loss_scale_decrease * disable amp outputs for checkpoint tests * add test for amp.state_dict, cleanup * add state_dict patch, add test * fixed testing, cleanup * add readme for checkpointing * add docs to source/amp * add review changes to doc
-
- 13 Aug, 2019 1 commit
-
-
Marek Kolodziej authored
Co-authored-by:
Aditya Agrawal <aditya.iitb@gmail.com> Co-authored-by:
Marek Kolodziej <mkolod@gmail.com>
-
- 24 Jun, 2019 1 commit
-
-
mcarilli authored
-
- 09 May, 2019 1 commit
-
-
Tim Zaman authored
-
- 30 Apr, 2019 1 commit
-
-
Michael Carilli authored
-
- 18 Apr, 2019 1 commit
-
-
Glenn Jocher authored
-
- 11 Apr, 2019 1 commit
-
-
Michael Carilli authored
-
- 12 Mar, 2019 1 commit
-
-
mcarilli authored
-
- 07 Mar, 2019 2 commits
-
-
Michael Carilli authored
-
Michael Carilli authored
-
- 04 Mar, 2019 1 commit
-
-
Michael Carilli authored
-
- 01 Mar, 2019 2 commits
- 28 Feb, 2019 1 commit
-
-
vfdev authored
-
- 20 Feb, 2019 5 commits
- 28 Jan, 2019 1 commit
-
-
mcarilli authored
-
