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: 812357d63c

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 '812357d63c'

${ noResults }

ColossalAI/tests

History

Jiarui Fang 569357fea0 add pytorch hooks (#179) ... * add pytorch hooks fix #175 * remove licenses in src code * add gpu memory tracer * replacing print with logger in ophooks.		3 years ago
..
test_comm	Hotfix/Colossalai layers (#92)	3 years ago
test_config	add pytorch hooks (#179)	3 years ago
test_context	Optimize pipeline schedule (#94)	3 years ago
test_data	added CI for unit testing (#69)	3 years ago
test_data_pipeline_tensor_parallel	Optimize pipeline schedule (#94)	3 years ago
test_engine/test_engine	Hotfix/Colossalai layers (#92)	3 years ago
test_layers	adapted for sequence parallel (#163)	3 years ago
test_trainer	pipeline last stage supports multi output (#151)	3 years ago
test_utils	Hotfix/Colossalai layers (#92)	3 years ago
test_zero_data_parallel	Hotfix/Colossalai layers (#92)	3 years ago
test_zero_tensor_parallel	Hotfix/Colossalai layers (#92)	3 years ago