/
githubmirror
/
ColossalAI
ОбзорДокументацияВойти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
ColossalAI/
.../nn/layer/
..
colossalai_layer

added skip_bias_add for non-tp linear

4 года назад
moe

[setup] support pre-build and jit-build of cuda kernels (#2374)

4 года назад
parallel_1d

[tensorparallel] fixed tp layers (#1938)

4 года назад
parallel_2d

[utils] refactor parallel layers checkpoint and bcast model on loading checkpoint (#1548)

4 года назад
parallel_2p5d

[utils] refactor parallel layers checkpoint and bcast model on loading checkpoint (#1548)

4 года назад
parallel_3d

improved allgather & reducescatter for 3d

4 года назад
parallel_sequence

[NFC] polish colossalai/nn/layer/parallel_sequence/layers.py code style (#1280)

4 года назад
utils

[NFC] polish colossalai/nn/layer/utils/common.py code style (#983)

4 года назад
vanilla

added skip_bias_add for non-tp linear

4 года назад
wrapper

[NFC] polish colossalai/nn/layer/wrapper/pipeline_wrapper.py code style (#1303)

4 года назад
__init__.py

[MOE] changed parallelmode to dist process group (#460)

4 года назад
base_layer.py

[utils] refactor parallel layers checkpoint and bcast model on loading checkpoint (#1548)

4 года назад
ДокументацияПоддержка
Политика конфиденциальностиПользовательское соглашениеПолитика использования «cookies»Согласие субъекта персональных данных
2026 ©