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: cloud/coati

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 'cloud/coati'

${ noResults }

ColossalAI/colossalai

History

Xuanlei Zhao 32e7f99416 [kernel] update triton init #4740 (#4740)		1 year ago
..
_C	[setup] support pre-build and jit-build of cuda kernels (#2374)	2 years ago
_analyzer	[example] add train resnet/vit with booster example (#3694)	2 years ago
amp	[pipeline] support fp32 for HybridPlugin/merge shardformer test and pipeline test into one file (#4354)	1 year ago
auto_parallel	[legacy] move communication and nn to legacy and refactor logger (#4671)	1 year ago
autochunk	fix typo colossalai/auto_parallel autochunk fx/passes etc. (#3808)	2 years ago
booster	[shardformer] add custom policy in hybrid parallel plugin (#4718)	1 year ago
checkpoint_io	[example] llama2 add fine-tune example (#4673)	1 year ago
cli	[legacy] move communication and nn to legacy and refactor logger (#4671)	1 year ago
cluster	[shardformer] support interleaved pipeline (#4448)	1 year ago
context	[legacy] move builder and registry to legacy (#4603)	1 year ago
device	fix some typo with colossalai/device colossalai/tensor/ etc. (#4171)	1 year ago
fx	[nfc] fix typo colossalai/cli fx kernel (#3847)	1 year ago
inference	[hotfix] Fix import error: colossal.kernel without triton installed (#4722)	1 year ago
interface	[zero] hotfix master param sync (#4618)	1 year ago
kernel	[kernel] update triton init #4740 (#4740)	1 year ago
lazy	[shardformer] support lazy init (#4202)	1 year ago
legacy	[legacy] move communication and nn to legacy and refactor logger (#4671)	1 year ago
logging	[legacy] move communication and nn to legacy and refactor logger (#4671)	1 year ago
nn	[legacy] move communication and nn to legacy and refactor logger (#4671)	1 year ago
pipeline	[legacy] move communication and nn to legacy and refactor logger (#4671)	1 year ago
shardformer	[doc] Add user document for Shardformer (#4702)	1 year ago
tensor	fix some typo with colossalai/device colossalai/tensor/ etc. (#4171)	1 year ago
testing	Next commit [checkpointio] Unsharded Optimizer Checkpoint for Gemini Plugin (#4141)	1 year ago
utils	[legacy] move communication and nn to legacy and refactor logger (#4671)	1 year ago
zero	[legacy] move communication and nn to legacy and refactor logger (#4671)	1 year ago
__init__.py	[setup] supported conda-installed torch (#2048)	2 years ago
constants.py	updated tp layers	2 years ago
core.py	[Tensor] distributed view supports inter-process hybrid parallel (#1169)	2 years ago
global_variables.py	[NFC] polish colossalai/global_variables.py code style (#3259)	2 years ago
initialize.py	[legacy] move builder and registry to legacy (#4603)	1 year ago