ColossalAI/colossalai
Frank Lee 551cafec14
[doc] updated kernel-related optimisers' docstring (#2385)
* [doc] updated kernel-related optimisers' docstring

* polish doc
2023-01-09 17:13:53 +08:00
..
_C [setup] support pre-build and jit-build of cuda kernels (#2374) 2023-01-06 20:50:26 +08:00
amp [setup] support pre-build and jit-build of cuda kernels (#2374) 2023-01-06 20:50:26 +08:00
auto_parallel Revert "[NFC] polish code format" (#2372) 2023-01-06 16:01:09 +08:00
builder
cli [cli] updated installation check cli for aot/jit build (#2395) 2023-01-09 11:05:27 +08:00
communication [NFC] polish communication/p2p_v2.py code style (#2303) 2023-01-04 15:09:57 +08:00
context
device [device] find best logical mesh 2023-01-05 17:21:29 +08:00
engine
fx [auto-parallel] refactoring ColoTracer (#2118) 2023-01-04 14:44:22 +08:00
gemini
kernel [hotfix] issue #2388 2023-01-07 18:23:02 +08:00
logging
nn [doc] updated kernel-related optimisers' docstring (#2385) 2023-01-09 17:13:53 +08:00
pipeline fix diff device in some partition 2023-01-06 15:59:06 +08:00
registry
tensor [hotfix] fix implement error in diffusers 2023-01-07 07:56:39 +08:00
testing [amp] add gradient clipping for unit tests (#2283) 2023-01-04 11:59:56 +08:00
trainer
utils [setup] support pre-build and jit-build of cuda kernels (#2374) 2023-01-06 20:50:26 +08:00
zero
__init__.py
constants.py
core.py
global_variables.py
initialize.py