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: 152162a80e

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 '152162a80e'

${ noResults }

ColossalAI/colossalai

History

Hongxin Liu cf519dac6a [optim] hotfix adam load (#6146) ... * [optim] hotfix adam load * [checkpointio] fix optimizer async io * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * [checkpointio] update test * [checkpointio] update test --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>		2 days ago
..
_C	Clean up	6 months ago
_analyzer	[test] Fix/fix testcase (#5770)	6 months ago
accelerator	[misc] fit torch api upgradation and remove legecy import (#6093)	1 month ago
amp	[Zerobubble] merge main. (#6142)	3 days ago
auto_parallel	[pre-commit.ci] pre-commit autoupdate (#5572)	5 months ago
autochunk	[hotfix] Fix examples no pad token & auto parallel codegen bug; (#5606)	7 months ago
booster	[optim] hotfix adam load (#6146)	2 days ago
checkpoint_io	[async io]supoort async io (#6137)	3 days ago
cli	[cli] support run as module option (#6135)	1 week ago
cluster	[FP8] rebase main (#5963)	4 months ago
context	[Fix]: implement thread-safety singleton to avoid deadlock for very large-scale training scenarios (#5625)	7 months ago
device	[Feature] Distributed optimizers: Lamb, Galore, CAME and Adafactor (#5694)	6 months ago
fx	[test] Fix/fix testcase (#5770)	6 months ago
inference	[shardformer] fix linear 1d row and support uneven splits for fused qkv linear (#6084)	1 month ago
interface	[Zerobubble] merge main. (#6142)	3 days ago
kernel	[misc] fit torch api upgradation and remove legecy import (#6093)	1 month ago
lazy	[fp8] Merge feature/fp8_comm to main branch of Colossalai (#6016)	3 months ago
legacy	[fp8] Merge feature/fp8_comm to main branch of Colossalai (#6016)	3 months ago
logging	[fp8] Merge feature/fp8_comm to main branch of Colossalai (#6016)	3 months ago
moe	[hotfix] moe hybrid parallelism benchmark & follow-up fix (#6048)	2 months ago
nn	[optim] hotfix adam load (#6146)	2 days ago
pipeline	[Zerobubble] merge main. (#6142)	3 days ago
quantization	[fp8] add fallback and make compile option configurable (#6092)	1 month ago
shardformer	[Zerobubble] merge main. (#6142)	3 days ago
tensor	[fp8] support fp8 amp for hybrid parallel plugin (#5975)	4 months ago
testing	[optim] hotfix adam load (#6146)	2 days ago
utils	[optim] hotfix adam load (#6146)	2 days ago
zero	[Zerobubble] merge main. (#6142)	3 days ago
__init__.py	[devops] remove post commit ci (#5566)	8 months ago
initialize.py	[fp8] hotfix backward hook (#6053)	2 months ago