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: feature/async-io

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 'feature/async-io'

${ noResults }

ColossalAI/colossalai/shardformer/policies

History

Hongxin Liu dc2cdaf3e8 [shardformer] optimize seq parallelism (#6086) ... * [shardformer] optimize seq parallelism * [shardformer] fix gpt2 fused linear col * [plugin] update gemini plugin * [plugin] update moe hybrid plugin * [test] update gpt2 fused linear test * [shardformer] fix gpt2 fused linear reduce		1 month ago
..
__init__.py	[shardformer] init shardformer code structure (#3731)	1 year ago
auto_policy.py	[FP8] rebase main (#5963)	4 months ago
base_policy.py	[fp8] Merge feature/fp8_comm to main branch of Colossalai (#6016)	3 months ago
bert.py	[shardformer] optimize seq parallelism (#6086)	1 month ago
blip2.py	[shardformer] fix linear 1d row and support uneven splits for fused qkv linear (#6084)	1 month ago
bloom.py	[shardformer] optimize seq parallelism (#6086)	1 month ago
chatglm2.py	[shardformer] optimize seq parallelism (#6086)	1 month ago
command.py	[fp8] Merge feature/fp8_comm to main branch of Colossalai (#6016)	3 months ago
deepseek.py	[hotfix] moe hybrid parallelism benchmark & follow-up fix (#6048)	2 months ago
falcon.py	[fp8] support hybrid parallel plugin (#5982)	3 months ago
gpt2.py	[shardformer] optimize seq parallelism (#6086)	1 month ago
gptj.py	[shardformer] optimize seq parallelism (#6086)	1 month ago
llama.py	[Feature] Split cross-entropy computation in SP (#5959)	2 months ago
mistral.py	[fp8] Merge feature/fp8_comm to main branch of Colossalai (#6016)	3 months ago
mixtral.py	[fp8] fix missing fp8_comm flag in mixtral (#6057)	2 months ago
opt.py	[fp8] support hybrid parallel plugin (#5982)	3 months ago
qwen2.py	[Feature] Zigzag Ring attention (#5905)	3 months ago
sam.py	[shardformer] fix linear 1d row and support uneven splits for fused qkv linear (#6084)	1 month ago
t5.py	[fp8] support hybrid parallel plugin (#5982)	3 months ago
vit.py	[fp8] support hybrid parallel plugin (#5982)	3 months ago
whisper.py	[fp8] support hybrid parallel plugin (#5982)	3 months ago