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

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

${ noResults }

ColossalAI/colossalai/shardformer/modeling

History

Tong Li 4c8e85ee0d [Coati] Train DPO using PP (#6054) ... * update dpo * remove unsupport plugin * update msg * update dpo * remove unsupport plugin * update msg * update template * update dataset * add pp for dpo * update dpo * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * add dpo fn * update dpo * update dpo * update dpo * update dpo * minor update * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * update loss * update help * polish code --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>		1 month ago
..
chatglm2_6b	[pre-commit.ci] pre-commit autoupdate (#5572)	5 months ago
__init__.py	[shardformer] added development protocol for standardization (#4149)	1 year ago
bert.py	[fp8] support hybrid parallel plugin (#5982)	3 months ago
blip2.py	[shardformer] support bias_gelu_jit_fused for models (#5647)	7 months ago
bloom.py	[Feature] Split cross-entropy computation in SP (#5959)	2 months ago
chatglm2.py	[Feature] Split cross-entropy computation in SP (#5959)	2 months ago
command.py	[Feature] Split cross-entropy computation in SP (#5959)	2 months ago
deepseek.py	[moe] add parallel strategy for shared_expert && fix test for deepseek (#6063)	2 months ago
falcon.py	[shardformer] fix modeling of bloom and falcon (#5796)	5 months ago
gpt2.py	[Feature] Split cross-entropy computation in SP (#5959)	2 months ago
gptj.py	[fp8] support hybrid parallel plugin (#5982)	3 months ago
jit.py	[misc] update pre-commit and run all files (#4752)	1 year ago
llama.py	[Coati] Train DPO using PP (#6054)	1 month ago
mistral.py	[Feature] Split cross-entropy computation in SP (#5959)	2 months ago
mixtral.py	[fp8] fix missing fp8_comm flag in mixtral (#6057)	2 months ago
opt.py	[Feature] Split cross-entropy computation in SP (#5959)	2 months ago
qwen2.py	[Feature] Split cross-entropy computation in SP (#5959)	2 months ago
sam.py	[shardformer]delete xformers (#5859)	5 months ago
t5.py	[shardformer] Support the T5ForTokenClassification model (#5816)	5 months ago
vit.py	[shardformer] support bias_gelu_jit_fused for models (#5647)	7 months ago
whisper.py	[shardformer] upgrade transformers to 4.39.3 (#5815)	5 months ago