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: refactor/inference

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 'refactor/inference'

${ noResults }

ColossalAI/tests

History

Yuanheng Zhao 916459c99a [inference] Add model forward accuracy test (#5102) ... * [inference] add model forward accuracy test * trivial: modify params * trivial: remove print message		1 year ago
..
kit	[Inference] Dynamic Batching Inference, online and offline (#4953)	1 year ago
test_analyzer	[misc] update pre-commit and run all files (#4752)	1 year ago
test_auto_parallel	[misc] update pre-commit and run all files (#4752)	1 year ago
test_autochunk	[misc] update pre-commit and run all files (#4752)	1 year ago
test_booster	[npu] add npu support for gemini and zero (#5067)	1 year ago
test_checkpoint_io	[gemini] gemini support extra-dp (#5043)	1 year ago
test_cluster	[misc] update pre-commit and run all files (#4752)	1 year ago
test_config	[misc] update pre-commit and run all files (#4752)	1 year ago
test_device	[misc] update pre-commit and run all files (#4752)	1 year ago
test_fx	[misc] update pre-commit and run all files (#4752)	1 year ago
test_gptq	[feature] add gptq for inference (#4754)	1 year ago
test_infer	[inference] Add model forward accuracy test (#5102)	1 year ago
test_infer_ops/triton	[inference] Refactor inference architecture (#5057)	1 year ago
test_lazy	[lazy] support from_pretrained (#4801)	1 year ago
test_legacy	[npu] add npu support for gemini and zero (#5067)	1 year ago
test_moe	[hotfix]: modify create_ep_hierarchical_group and add test (#5032)	1 year ago
test_optimizer	[test] merge old components to test to model zoo (#4945)	1 year ago
test_pipeline	[misc] update pre-commit and run all files (#4752)	1 year ago
test_shardformer	[gemini] gemini support extra-dp (#5043)	1 year ago
test_smoothquant	[inference] Add smmoothquant for llama (#4904)	1 year ago
test_tensor	[misc] update pre-commit and run all files (#4752)	1 year ago
test_utils	[misc] update pre-commit and run all files (#4752)	1 year ago
test_zero	[npu] add npu support for gemini and zero (#5067)	1 year ago
__init__.py	[zero] Update sharded model v2 using sharded param v2 (#323)	3 years ago