ColossalAI/colossalai
Jiarui Fang ab95ec9aea
[Tensor] init ColoParameter (#914)
2022-05-06 12:57:14 +08:00
..
amp [doc] improved docstring in the amp module (#857) 2022-04-25 13:42:17 +08:00
builder
cli [cli] refactored micro-benchmarking cli and added more metrics (#858) 2022-04-25 11:48:07 +08:00
communication [doc] improved docstring in the communication module (#863) 2022-04-25 13:41:43 +08:00
context [usability] improved error messages in the context module (#856) 2022-04-25 13:42:31 +08:00
engine [doc] improved docstring and assertion messages for the engine module (#871) 2022-04-26 10:00:18 +08:00
gemini [gemini] accelerate adjust_layout() (#878) 2022-04-26 18:08:31 +08:00
kernel
logging [doc] improved docstring in the logging module (#861) 2022-04-25 13:42:00 +08:00
nn
registry
tensor [Tensor] init ColoParameter (#914) 2022-05-06 12:57:14 +08:00
testing
trainer [doc] improved assertion messages in trainer (#873) 2022-04-26 10:00:12 +08:00
utils [Tensor] init ColoParameter (#914) 2022-05-06 12:57:14 +08:00
zero [gemini] accelerate adjust_layout() (#878) 2022-04-26 18:08:31 +08:00
__init__.py
constants.py
core.py
global_variables.py
initialize.py [doc] improved error messages in initialize (#872) 2022-04-26 10:00:03 +08:00