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: 279300dc5f

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 '279300dc5f'

${ noResults }

ColossalAI/colossalai

History

Yuanheng Zhao 04863a9b14 [example] Update Llama Inference example (#5629) ... * [example] add infernece benchmark llama3 * revise inference config - arg * remove unused args * add llama generation demo script * fix init rope in llama policy * add benchmark-llama3 - cleanup		7 months ago
..
_C	…
_analyzer	[hotfix] quick fixes to make legacy tutorials runnable (#5559)	8 months ago
accelerator	[hotfix] fix typo change MoECheckpintIO to MoECheckpointIO (#5335)	9 months ago
amp	[npu] change device to accelerator api (#5239)	11 months ago
auto_parallel	[hotfix] Fix wrong import in meta_registry (#5392)	9 months ago
autochunk	…
booster	[devops] remove post commit ci (#5566)	8 months ago
checkpoint_io	[devops] remove post commit ci (#5566)	8 months ago
cli	[devops] fix extention building (#5427)	9 months ago
cluster	[devops] remove post commit ci (#5566)	8 months ago
context	…
device	…
fx	…
inference	[example] Update Llama Inference example (#5629)	7 months ago
interface	…
kernel	[Fix/Inference] Fix GQA Triton and Support Llama3 (#5624)	7 months ago
lazy	…
legacy	[Fix] resolve conflicts of merging main	8 months ago
logging	…
moe	[hotfix] fix typo change MoECheckpintIO to MoECheckpointIO (#5335)	9 months ago
nn	[hotfix] quick fixes to make legacy tutorials runnable (#5559)	8 months ago
pipeline	[shardformer, pipeline] add `gradient_checkpointing_ratio` and heterogenous shard policy for llama (#5508)	8 months ago
shardformer	[Fix] resolve conflicts of merging main	8 months ago
tensor	[devops] remove post commit ci (#5566)	8 months ago
testing	[shardformer] update colo attention to support custom mask (#5510)	8 months ago
utils	Merge pull request #5310 from hpcaitech/feature/npu	10 months ago
zero	[devops] remove post commit ci (#5566)	8 months ago
__init__.py	[devops] remove post commit ci (#5566)	8 months ago
initialize.py	[npu] change device to accelerator api (#5239)	11 months ago