ColossalAI/colossalai/communication
ver217 8f02a88db2
add interleaved pipeline, fix naive amp and update pipeline model initializer (#80)
2021-12-20 23:26:19 +08:00
..
__init__.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
collective.py update examples and sphnix docs for the new api (#63) 2021-12-13 22:07:01 +08:00
p2p.py add interleaved pipeline, fix naive amp and update pipeline model initializer (#80) 2021-12-20 23:26:19 +08:00
ring.py update examples and sphnix docs for the new api (#63) 2021-12-13 22:07:01 +08:00
utils.py add interleaved pipeline, fix naive amp and update pipeline model initializer (#80) 2021-12-20 23:26:19 +08:00