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: c9cff7e7fa

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 'c9cff7e7fa'

${ noResults }

ColossalAI/colossalai

History

Baizhou Zhang c9cff7e7fa [checkpointio] General Checkpointing of Sharded Optimizers (#3984)		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	[bf16] add bf16 support (#3882)	1 year ago
auto_parallel	[nfc] fix typo colossalai/ applications/ (#3831)	2 years ago
autochunk	fix typo colossalai/auto_parallel autochunk fx/passes etc. (#3808)	2 years ago
booster	[checkpointio] General Checkpointing of Sharded Optimizers (#3984)	1 year ago
builder	…
checkpoint_io	[checkpointio] General Checkpointing of Sharded Optimizers (#3984)	1 year ago
cli	Modify torch version requirement to adapt torch 2.0 (#3896)	1 year ago
cluster	fix typo colossalai/auto_parallel autochunk fx/passes etc. (#3808)	2 years ago
communication	[CI] fix some spelling errors (#3707)	2 years ago
context	[CI] fix some spelling errors (#3707)	2 years ago
device	Revert "[sync] sync feature/shardformer with develop"	1 year ago
engine	fix typo colossalai/auto_parallel autochunk fx/passes etc. (#3808)	2 years ago
fx	[nfc] fix typo colossalai/cli fx kernel (#3847)	1 year ago
interface	[booster] implemented the torch ddd + resnet example (#3232)	2 years ago
kernel	[bf16] add bf16 support (#3882)	1 year ago
lazy	Revert "[sync] sync feature/shardformer with develop"	1 year ago
logging	[logger] hotfix, missing _FORMAT (#2231)	2 years ago
nn	Revert "[sync] sync feature/shardformer with develop"	1 year ago
pipeline	[nfc]fix typo colossalai/pipeline tensor nn (#3899)	1 year ago
registry	…
tensor	Revert "[sync] sync feature/shardformer with develop"	1 year ago
testing	[NFC] fix typo with colossalai/auto_parallel/tensor_shard (#3742)	2 years ago
trainer	fix typo with colossalai/trainer utils zero (#3908)	1 year ago
utils	fix typo with colossalai/trainer utils zero (#3908)	1 year ago
zero	[gemini] fixed the gemini checkpoint io (#3934)	1 year ago
__init__.py	[setup] supported conda-installed torch (#2048)	2 years ago
constants.py	…
core.py	…
global_variables.py	[NFC] polish colossalai/global_variables.py code style (#3259)	2 years ago
initialize.py	[nfc] fix typo colossalai/zero (#3923)	1 year ago