ColossalAI/colossalai/nn/layer/parallel_2p5d
Liang Bowen ec5086c49c Refactored docstring to google style 2022-03-29 17:17:47 +08:00
..
__init__.py moved env variables to global variables; (#215) 2022-02-15 11:31:13 +08:00
_operation.py Refactored docstring to google style 2022-03-29 17:17:47 +08:00
_utils.py fix format parallel_2p5d (#357) 2022-03-11 15:50:28 +08:00
layers.py Refactored docstring to google style 2022-03-29 17:17:47 +08:00