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: 191daf7411

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 '191daf7411'

${ noResults }

ColossalAI/colossalai

History

Frank Lee 95a36eae63 [kernel] added kernel loader to softmax autograd function (#3093) ... * [kernel] added kernel loader to softmax autograd function * [release] v0.2.6		2 years ago
..
_C	[setup] support pre-build and jit-build of cuda kernels (#2374)	2 years ago
_analyzer	[analyzer] a minimal implementation of static graph analyzer (#2852)	2 years ago
amp	[test] fixed the triton version for testing (#2608)	2 years ago
auto_parallel	[hotfix] add shard dim to aviod backward communication error (#2954)	2 years ago
autochunk	[autochunk] support vit (#3084)	2 years ago
booster	[booster] init module structure and definition (#3056)	2 years ago
builder	…
cli	[cli] handled version check exceptions (#2848)	2 years ago
communication	[NFC] polish communication/p2p_v2.py code style (#2303)	2 years ago
context	[NFC] polish colossalai/context/process_group_initializer/initializer_2d.py code style (#2726)	2 years ago
device	[hotfix] add copyright for solver and device mesh (#2803)	2 years ago
engine	[doc] add deepspeed citation and copyright (#2996)	2 years ago
fx	[analyzer] a minimal implementation of static graph analyzer (#2852)	2 years ago
gemini	[doc] add deepspeed citation and copyright (#2996)	2 years ago
kernel	[kernel] added kernel loader to softmax autograd function (#3093)	2 years ago
logging	[logger] hotfix, missing _FORMAT (#2231)	2 years ago
nn	[format] applied code formatting on changed files in pull request 2997 (#3008)	2 years ago
pipeline	[pipeline] Add Simplified Alpa DP Partition (#2507)	2 years ago
registry	…
tensor	[DTensor] implement layout converter (#3055)	2 years ago
testing	[amp] add gradient clipping for unit tests (#2283)	2 years ago
trainer	[polish] remove useless file _mem_tracer_hook.py (#1963)	2 years ago
utils	[doc] add deepspeed citation and copyright (#2996)	2 years ago
zero	[doc] add deepspeed citation and copyright (#2996)	2 years ago
__init__.py	[setup] supported conda-installed torch (#2048)	2 years ago
constants.py	updated tp layers	2 years ago
core.py	…
global_variables.py	updated tp layers	2 years ago
initialize.py	Fix False warning in initialize.py (#2456)	2 years ago