Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
COMMITS
/ tests/plugins/test_amp_plugins.py April 20, 2022
A
Make standalone tests less verbose (#12684)
Adrian Wälchli committed
February 21, 2022
J
Update `tests/plugins/*.py` to use `devices` instead of `gpus` or `ipus` (#11872)
Jv Kyle Eclarin committed
February 9, 2022
A
Add `Accelerator.is_available()` interface requirement (#11797)
ananthsub committed
November 26, 2021
C
Rename special to standalone (#10779)
Carlos Mocholí committed
November 17, 2021
C
Fail the test when a `DeprecationWarning` is raised (#9940)
Carlos Mocholí committed
October 30, 2021
C
Clip before step (#10248)
Carlos Mocholí committed
October 28, 2021
C
Fix gradient norm tracking and gradient clipping (#9287)
Carlos Mocholí committed
C
Pass the scaler as an input to `NativeMixedPrecisionPlugin` (#10055)
Carlos Mocholí committed
October 27, 2021
C
Replace `_TORCH_GREATER_EQUAL_DEV_1_10` with `_TORCH_GREATER_EQUAL_1_10` (#10157)
Carlos Mocholí committed
October 25, 2021
C
Use `torch.autocast` (#10053)
Carlos Mocholí committed
October 20, 2021
C
Remove unnecessary dependency available checks (#10050)
Carlos Mocholí committed
C
Default to `precision=bf16` on CPU when `precision=16` is passed (#10033)
Carlos Mocholí committed
October 16, 2021
K
October 4, 2021
D
Fix broken `test_cpu_amp_precision_context_manager` (#9809)
Danielle Pintz committed
September 29, 2021
C
Remove `_NATIVE_AMP_AVAILABLE` checks (#9747)
Carlos Mocholí committed
September 6, 2021
J
CI: precommit - docformatter (#8584)
Jirka Borovec committed
September 4, 2021
J
Update call to `amp.autocast` from `fast_dtype` to `dtype` (#9211)
John St. John committed
August 25, 2021
S
Add support for CPU AMP autocast (#9084)
Sean Naren committed
August 24, 2021
S
Add bfloat16 support to Lightning Trainer (#9049)
Sean Naren committed
July 26, 2021
C
Replace `yapf` with `black` (#7783)
Carlos Mocholí committed
July 9, 2021
D
Add the `on_before_optimizer_step` hook (#8048)
Dusan Drevicky committed
July 8, 2021
C
Refactor plugins backward (#8328)
Carlos Mocholí committed
June 16, 2021
Y
Make optimizers skippable when using amp (#7975)
Yifu Wang committed
April 30, 2021
T
[bugfix] Apex never instantiated. (#7274)
thomas chaton committed
April 13, 2021
A
Clean up environment access in plugins (#6941)
Adrian Wälchli committed
March 3, 2021
J
Simplify test for AMP plugins (#6311)
Jirka Borovec committed