ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-27 02:03:13 +00:00

Files

History

HELSON 0f8c7f9804 Fixed docstring in colossalai (#171 )

2022-01-21 10:44:30 +08:00

__init__.py

Added MoE parallel (#127 )

2022-01-07 15:08:36 +08:00

initializer_1d.py

2022-01-21 10:44:30 +08:00

initializer_2d.py

2022-01-21 10:44:30 +08:00

initializer_2p5d.py

2022-01-21 10:44:30 +08:00

initializer_3d.py

2022-01-21 10:44:30 +08:00

initializer_data.py

2022-01-21 10:44:30 +08:00

initializer_model.py

2022-01-21 10:44:30 +08:00

initializer_moe.py

2022-01-21 10:44:30 +08:00

initializer_pipeline.py

2022-01-21 10:44:30 +08:00

initializer_sequence.py

2022-01-21 10:44:30 +08:00

initializer_tensor.py

2022-01-21 10:44:30 +08:00

process_group_initializer.py

2022-01-21 10:44:30 +08:00