github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
Making large AI models cheaper, faster and more accessible
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.
562 Commits
31 Branches
46 Tags
236 MiB
 
 
 
 
 
Tree: 193d629311
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from '193d629311'
${ noResults }
ColossalAI/colossalai
Ziyue Jiang f593a5637e
[Tensor] add embedding tp1d row (#904) 		3 years ago
..
amp [doc] improved docstring in the amp module (#857) 3 years ago
builder
…
cli [cli] refactored micro-benchmarking cli and added more metrics (#858) 3 years ago
communication [doc] improved docstring in the communication module (#863) 3 years ago
context [usability] improved error messages in the context module (#856) 3 years ago
engine [doc] improved docstring and assertion messages for the engine module (#871) 3 years ago
gemini [gemini] accelerate adjust_layout() (#878) 3 years ago
kernel
…
logging [doc] improved docstring in the logging module (#861) 3 years ago
nn
…
registry
…
tensor [Tensor] add embedding tp1d row (#904) 3 years ago
testing
…
trainer [doc] improved assertion messages in trainer (#873) 3 years ago
utils [Tensor] initialize the ColoOptimizer (#898) 3 years ago
zero [gemini] accelerate adjust_layout() (#878) 3 years ago
__init__.py
…
constants.py
…
core.py
…
global_variables.py
…
initialize.py [doc] improved error messages in initialize (#872) 3 years ago