Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
COMMITS
/ docs/source-pytorch/extensions/plugins.rst October 30, 2023
A
Rename PrecisionPlugin -> Precision (#18840)
Adrian Wälchli committed
September 29, 2023
C
Bitsandbytes precision plugin (#18655)
Carlos Mocholí committed
September 8, 2023
A
Change auto-device selection for Jupyter notebook environments (#18291)
Adrian Wälchli committed
September 7, 2023
C
Support the TransformerEngine precision plugin with the Trainer (#18459)
Carlos Mocholí committed
August 10, 2023
A
Support true 16-bit precision with FSDP in Trainer (#18219)
Adrian Wälchli committed
April 28, 2023
C
[TPU] Rename classes to use XLA instead of TPU (#17383)
Carlos Mocholí committed
April 27, 2023
J
Replace IPU with external implementation (#17075)
Jirka Borovec committed
March 24, 2023
J
Replace HPU with external implementation (#17067)
Jirka Borovec committed
February 27, 2023
J
docs: update `pytorch_lightning` imports (#16864)
Jirka Borovec committed
February 15, 2023
A
Replace ColossalAIStrategy with external implementation (#16757)
Adrian Wälchli committed
January 25, 2023
C
Remove the "native" suffix from the codebase (#16490)
Carlos Mocholí committed
January 23, 2023
C
Remove the FairScale integration (#16400)
Carlos Mocholí committed
December 20, 2022
C
Deprecate nvidia/apex (#16039)
Carlos Mocholí committed
October 11, 2022
V
Add ColossalAI strategy (#14224)
ver217 committed
September 22, 2022
A
Integrate Lite Precision into PL (#14798)
Adrian Wälchli committed
August 13, 2022
R
Use fsdp module to initialize precision scalar for fsdp native (#14092)
Rohit Gupta committed
July 26, 2022
R
Add support for async checkpointing (#13658)
Rohit Gupta committed
June 22, 2022
A
Add `XLAEnvironment` plugin (#11330)
Adrian Wälchli committed
June 15, 2022
J
Future 3/n: docs adjustment (#13299)
Jirka Borovec committed