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: 1c43bfd54e

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 '1c43bfd54e'

${ noResults }

ColossalAI/colossalai

History

Hongxin Liu 0b00def881 [example] add llama2 example (#4527) ... * [example] transfer llama-1 example * [example] fit llama-2 * [example] refactor scripts folder * [example] fit new gemini plugin * [cli] fix multinode runner * [example] fit gemini optim checkpoint * [example] refactor scripts * [example] update requirements * [example] update requirements * [example] rename llama to llama2 * [example] update readme and pretrain script * [example] refactor scripts		1 year ago
..
_C	…
_analyzer	…
amp	[pipeline] support fp32 for HybridPlugin/merge shardformer test and pipeline test into one file (#4354)	1 year ago
auto_parallel	[NFC] polish runtime_preparation_pass style (#4266)	1 year ago
autochunk	fix typo colossalai/auto_parallel autochunk fx/passes etc. (#3808)	2 years ago
booster	[gemini] improve compatibility and add static placement policy (#4479)	1 year ago
builder	…
checkpoint_io	[checkpointio] Sharded Optimizer Checkpoint for Gemini Plugin (#4302)	1 year ago
cli	[example] add llama2 example (#4527)	1 year ago
cluster	[cluster] add process group mesh (#4039)	1 year ago
communication	[NFC] fix: format (#4270)	1 year ago
context	…
device	[format] applied code formatting on changed files in pull request 4152 (#4157)	1 year ago
engine	[nfc]fix ColossalaiOptimizer is not defined (#4122)	1 year ago
fx	[nfc] fix typo colossalai/cli fx kernel (#3847)	1 year ago
interface	[pipeline] refactor 1f1b schedule (#4115)	1 year ago
kernel	[example] add llama2 example (#4527)	1 year ago
lazy	[shardformer] support lazy init (#4202)	1 year ago
logging	…
nn	[doc] add Series A Funding and NeurIPS news (#4377)	1 year ago
pipeline	[pipeline] rewrite t5 tests & support multi-tensor transmitting in pipeline (#4388)	1 year ago
registry	…
shardformer	[format] applied code formatting on changed files in pull request 4441 (#4445)	1 year ago
tensor	[gemini] improve compatibility and add static placement policy (#4479)	1 year ago
testing	Next commit [checkpointio] Unsharded Optimizer Checkpoint for Gemini Plugin (#4141)	1 year ago
trainer	fix typo with colossalai/trainer utils zero (#3908)	1 year ago
utils	[test] remove useless tests (#4359)	1 year ago
zero	[zero]support zero2 with gradient accumulation (#4511)	1 year ago
__init__.py	…
constants.py	…
core.py	…
global_variables.py	…
initialize.py	[nfc] fix typo colossalai/zero (#3923)	1 year ago