Making large AI models cheaper, faster and more accessible

aibig-modeldata-parallelismdeep-learningdistributed-computingfoundation-modelsheterogeneous-traininghpcinferencelarge-scalemodel-parallelismpipeline-parallelism

You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Branch: main

ColossalChat

ckpt

cloud/coati

colossalchat

colossalchat_upgrade

dev/gpt2_metainfo_patch

dev/zero-offload

develop

feat/moe

feat/online-serving

feat/speculative-decoding

feature/2-stage

feature/async-io

feature/colossal-infer

feature/elixir

feature/inference-refactor

feature/lora

feature/stable-diffusion

feature/zerobubble

fix-setup

fix/format

flybird11111-patch-1

hotfix/example_test

hotfix/kernel_build_before_load

llama3

main

moe_sp

pre-commit-ci-update-config

refactor/inference

supercooledith-patch-1

support-npu

v0.0.1-beta

v0.0.2

v0.1.0

v0.1.1

v0.1.10

v0.1.11rc1

v0.1.11rc2

v0.1.11rc3

v0.1.11rc4

v0.1.11rc5

v0.1.12

v0.1.13

v0.1.2

v0.1.3

v0.1.4

v0.1.5

v0.1.6

v0.1.7

v0.1.8

v0.1.9

v0.2.0

v0.2.1

v0.2.2

v0.2.3

v0.2.4

v0.2.5

v0.2.6

v0.2.7

v0.2.8

v0.3.0

v0.3.1

v0.3.2

v0.3.3

v0.3.4

v0.3.5

v0.3.6

v0.3.7

v0.3.8

v0.3.9

v0.4.0

v0.4.1

v0.4.2

v0.4.3

v0.4.4

v0.4.5

v0.4.6

Branches Tags

${ item.name }

Create tag ${ searchTerm }

Create branch ${ searchTerm }

from 'main'

${ noResults }

ColossalAI/tests

History

Hongxin Liu cf519dac6a [optim] hotfix adam load (#6146) ... * [optim] hotfix adam load * [checkpointio] fix optimizer async io * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * [checkpointio] update test * [checkpointio] update test --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>		1 day ago
..
kit	[Feature] Split cross-entropy computation in SP (#5959)	2 months ago
test_analyzer	[misc] update pre-commit and run all files (#4752)	1 year ago
test_auto_parallel	[test] Fix/fix testcase (#5770)	6 months ago
test_autochunk	[misc] update pre-commit and run all files (#4752)	1 year ago
test_booster	[plugin] support get_grad_norm (#6115)	2 weeks ago
test_checkpoint_io	[optim] hotfix adam load (#6146)	1 day ago
test_cluster	[misc] refactor launch API and tensor constructor (#5666)	7 months ago
test_config	[misc] update pre-commit and run all files (#4752)	1 year ago
test_device	[misc] refactor launch API and tensor constructor (#5666)	7 months ago
test_fp8	[fp8] Disable all_gather intranode. Disable Redundant all_gather fp8 (#6059)	2 months ago
test_fx	[hotfix] fix testcase in test_fx/test_tracer (#5779)	6 months ago
test_infer	[release] update version (#6041)	2 months ago
test_lazy	[fp8] Merge feature/fp8_comm to main branch of Colossalai (#6016)	3 months ago
test_legacy	[FP8] rebase main (#5963)	4 months ago
test_lora	[fp8] Merge feature/fp8_comm to main branch of Colossalai (#6016)	3 months ago
test_moe	[hotfix] moe hybrid parallelism benchmark & follow-up fix (#6048)	2 months ago
test_optimizer	[MoE/ZeRO] Moe refactor with zero refactor (#5821)	5 months ago
test_pipeline	[Zerobubble] merge main. (#6142)	2 days ago
test_shardformer	[Zerobubble] merge main. (#6142)	2 days ago
test_smoothquant	[inference] Add smmoothquant for llama (#4904)	1 year ago
test_tensor	[misc] refactor launch API and tensor constructor (#5666)	7 months ago
test_zero	[zero] support extra dp (#6123)	1 week ago
__init__.py	…