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

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

${ noResults }

ColossalAI/colossalai

History

Xu Kai 20332a7a34 [inference] udpate example (#5053) ... * udpate example * fix run.sh		1 year ago
..
_C	[setup] support pre-build and jit-build of cuda kernels (#2374)	2 years ago
_analyzer	[misc] update pre-commit and run all files (#4752)	1 year ago
amp	[feature] Add clip_grad_norm for hybrid_parallel_plugin (#4837)	1 year ago
auto_parallel	[misc] update pre-commit and run all files (#4752)	1 year ago
autochunk	[misc] update pre-commit and run all files (#4752)	1 year ago
booster	[moe] support optimizer checkpoint (#5015)	1 year ago
checkpoint_io	[checkpointio] hotfix torch 2.0 compatibility (#4824)	1 year ago
cli	[bug] Fix the version check bug in colossalai run when generating the cmd. (#4713)	1 year ago
cluster	[hotfix] Fix the bug where process groups were not being properly released. (#4940)	1 year ago
context	[moe] merge moe into main (#4978)	1 year ago
device	[misc] update pre-commit and run all files (#4752)	1 year ago
fx	[misc] update pre-commit and run all files (#4752)	1 year ago
inference	[inference] udpate example (#5053)	1 year ago
interface	[lazy] support from_pretrained (#4801)	1 year ago
kernel	[moe] merge moe into main (#4978)	1 year ago
lazy	[doc] add lazy init docs (#4808)	1 year ago
legacy	[inference] udpate example (#5053)	1 year ago
logging	[misc] update pre-commit and run all files (#4752)	1 year ago
moe	[moe] support optimizer checkpoint (#5015)	1 year ago
nn	[moe] merge moe into main (#4978)	1 year ago
pipeline	[Refactor] remove useless inference code (#5022)	1 year ago
shardformer	[Refactor] refactor policy search and quant type controlling in inference (#5035)	1 year ago
tensor	[moe] support optimizer checkpoint (#5015)	1 year ago
testing	[test] merge old components to test to model zoo (#4945)	1 year ago
utils	[moe] merge moe into main (#4978)	1 year ago
zero	[hotfix] Add layer norm gradients all-reduce for sequence parallel (#4926)	1 year ago
__init__.py	[misc] update pre-commit and run all files (#4752)	1 year ago
initialize.py	[misc] update pre-commit and run all files (#4752)	1 year ago