ColossalAI/colossalai/nn/layer/parallel_2d
HELSON 632e622de8
overlap computation and communication in 2d operations (#75)
2021-12-16 16:05:15 +08:00
..
__init__.py Migrated project 2021-10-28 18:21:23 +02:00
_operation.py overlap computation and communication in 2d operations (#75) 2021-12-16 16:05:15 +08:00
_transformer.py Migrated project 2021-10-28 18:21:23 +02:00
_utils.py Migrated project 2021-10-28 18:21:23 +02:00
_vit.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
layers.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00