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.
536 Commits
31 Branches
46 Tags
236 MiB
 
 
 
 
 
Tree: 2883040286
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from '2883040286'
${ noResults }
ColossalAI/colossalai
Jiarui Fang 96211c2cc8
[tensor] customized op returns ColoTensor (#875) 		3 years ago
..
amp [doc] improved docstring in the amp module (#857) 3 years ago
builder modefied the pp build for ckpt adaptation (#803) 3 years ago
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] add stateful tensor container (#867) 3 years ago
kernel
…
logging [doc] improved docstring in the logging module (#861) 3 years ago
nn [gemini] add GeminiMemoryManger (#832) 3 years ago
registry [usability] added assertion message in registry (#864) 3 years ago
tensor [tensor] customized op returns ColoTensor (#875) 3 years ago
testing
…
trainer [doc] improved assertion messages in trainer (#873) 3 years ago
utils colo init context add device attr. (#866) 3 years ago
zero [zero] use GeminiMemoryManager when sampling model data (#850) 3 years ago
__init__.py
…
constants.py
…
core.py
…
global_variables.py
…
initialize.py [doc] improved error messages in initialize (#872) 3 years ago