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

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

${ noResults }

ColossalAI/examples/language

History

botbw c54c4fcd15 [hotfix] moe hybrid parallelism benchmark & follow-up fix (#6048) ... * [example] pass use_fp8_comm flag to all plugins * [example] add mixtral benchmark * [moe] refine assertion and check * [moe] fix mixtral & add more tests * [moe] consider checking dp * sp group and moe_dp_group * [mixtral] remove gate tp & add more tests * [deepseek] fix tp & sp for deepseek * [mixtral] minor fix * [deepseek] add deepseek benchmark		2 months ago
..
bert	[Feature]: support FP8 communication in DDP, FSDP, Gemini (#5928)	4 months ago
commons	[example] make gpt example directory more clear (#2353)	2 years ago
deepseek	[hotfix] moe hybrid parallelism benchmark & follow-up fix (#6048)	2 months ago
gpt	[Feature] Split cross-entropy computation in SP (#5959)	2 months ago
grok-1	[misc] refactor launch API and tensor constructor (#5666)	7 months ago
llama	[hotfix] moe hybrid parallelism benchmark & follow-up fix (#6048)	2 months ago
mixtral	[hotfix] moe hybrid parallelism benchmark & follow-up fix (#6048)	2 months ago
opt	[fp8] Merge feature/fp8_comm to main branch of Colossalai (#6016)	3 months ago
palm	[misc] refactor launch API and tensor constructor (#5666)	7 months ago
__init__.py	[example]add gpt2 benchmark example script. (#5295)	9 months ago
data_utils.py	[devops] remove post commit ci (#5566)	8 months ago
model_utils.py	[example]add gpt2 benchmark example script. (#5295)	9 months ago
performance_evaluator.py	[Feature] Split cross-entropy computation in SP (#5959)	2 months ago