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: feature/2-stage

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 'feature/2-stage'

${ noResults }

ColossalAI/tests

History

YuliangLiu0306 5458da5c3c [autoparallel] integrate 2-stage solver (#3476) ... * integrate 2-stage solver * [autoparallel] integrate 2-stage solver * polish		1 year ago
..
components_to_test	[CI] fix typo with tests/ etc. (#3727)	2 years ago
kit	[shardformer] added embedding gradient check (#4124)	1 year ago
test_amp	[test] refactor tests with spawn (#3452)	2 years ago
test_analyzer	[devops] update torch version of CI (#3725)	2 years ago
test_auto_parallel	[autoparallel] integrate 2-stage solver (#3476)	1 year ago
test_autochunk	[test] fixed tests failed due to dtensor change (#4082)	1 year ago
test_booster	[shardformer] adapted T5 and LLaMa test to use kit (#4049)	1 year ago
test_checkpoint_io	[test] fixed tests failed due to dtensor change (#4082)	1 year ago
test_cluster	[devops] update torch version of CI (#3725)	2 years ago
test_comm	[test] refactor tests with spawn (#3452)	2 years ago
test_config	[pipeline] refactor the pipeline module (#1087)	2 years ago
test_context	[test] refactor tests with spawn (#3452)	2 years ago
test_data	[test] refactor tests with spawn (#3452)	2 years ago
test_data_pipeline_tensor_parallel	[CI] fix typo with tests/ etc. (#3727)	2 years ago
test_ddp	[test] refactor tests with spawn (#3452)	2 years ago
test_device	[format] applied code formatting on changed files in pull request 4152 (#4157)	1 year ago
test_engine	[test] refactor tests with spawn (#3452)	2 years ago
test_fx	[shardformer] shardformer support opt models (#4091)	1 year ago
test_layers	[CI] fix typo with tests/ etc. (#3727)	2 years ago
test_lazy	[test] fixed tests failed due to dtensor change (#4082)	1 year ago
test_moe	[CI] fix typo with tests/ etc. (#3727)	2 years ago
test_ops	[test] refactor tests with spawn (#3452)	2 years ago
test_optimizer	[bf16] add bf16 support (#3882)	1 year ago
test_pipeline	[test] refactor tests with spawn (#3452)	2 years ago
test_shardformer	[format] applied code formatting on changed files in pull request 4152 (#4157)	1 year ago
test_tensor	[test] fixed tests failed due to dtensor change (#4082)	1 year ago
test_trainer	[CI] fix typo with tests/ etc. (#3727)	2 years ago
test_utils	[lazy] refactor lazy init (#3891)	1 year ago
test_zero	[gemini] fix argument naming during chunk configuration searching	1 year ago
__init__.py	[zero] Update sharded model v2 using sharded param v2 (#323)	3 years ago