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.

Tree: 73b542a124

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 '73b542a124'

${ noResults }

ColossalAI/tests

History

Frank Lee 73d3e4d309 [booster] implemented the torch ddd + resnet example (#3232) ... * [booster] implemented the torch ddd + resnet example * polish code		2 years ago
..
components_to_test	[testing] add beit model for unit testings (#2196)	2 years ago
kit	[FX] refactor experimental tracer and adapt it with hf models (#3157)	2 years ago
test_amp	[workflow] only report coverage for changed files (#2524)	2 years ago
test_analyzer	[Analyzer] fix analyzer tests (#3197)	2 years ago
test_auto_parallel	[auto-parallel] add auto-offload feature (#3154)	2 years ago
test_autochunk	[autochunk] support complete benchmark (#3121)	2 years ago
test_booster	[booster] implemented the torch ddd + resnet example (#3232)	2 years ago
test_checkpoint_io	[booster] implemented the torch ddd + resnet example (#3232)	2 years ago
test_cluster	[API] implement device mesh manager (#3221)	2 years ago
test_comm	[communication] add p2p_v2.py to support communication with List[Any] (#1407)	2 years ago
test_config	[pipeline] refactor the pipeline module (#1087)	2 years ago
test_context	[test] refactored with the new rerun decorator (#763)	3 years ago
test_data	[unittest] refactored unit tests for change in dependency (#838)	3 years ago
test_data_pipeline_tensor_parallel	[engin/schedule] use p2p_v2 to recontruct pipeline_schedule (#1408)	2 years ago
test_ddp	[gemini] update ddp strict mode (#2518)	2 years ago
test_device	[device] find best logical mesh	2 years ago
test_engine	[hotfix] remove potiential circle import (#1307)	2 years ago
test_fx	[hotfix] skip torchaudio tracing test (#3211)	2 years ago
test_gemini	[test] fixed torchrec model test (#3167)	2 years ago
test_layers	improved allgather & reducescatter for 3d	2 years ago
test_moe	[test] align model name with the file name. (#2045)	2 years ago
test_ops	[FAW] export FAW in _ops (#1438)	2 years ago
test_optimizer	[setup] support pre-build and jit-build of cuda kernels (#2374)	2 years ago
test_pipeline	[PP Middleware] Add bwd and step for PP middleware (#2111)	2 years ago
test_tensor	[DTensor] refactor dtensor with new components (#3089)	2 years ago
test_trainer	[pipeline] refactor the pipeline module (#1087)	2 years ago
test_utils	[hotfix] skip torchaudio tracing test (#3211)	2 years ago
test_zero	[zero] add zero wrappers (#2523)	2 years ago
__init__.py	…