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.
ColossalAI/colossalai/legacy
Hongxin Liu 7f8b16635b
[misc] refactor launch API and tensor constructor (#5666)
7 months ago
..
amp [feat] refactored extension module (#5298) 10 months ago
builder
communication
context
engine
inference [misc] refactor launch API and tensor constructor (#5666) 7 months ago
nn [feat] refactored extension module (#5298) 10 months ago
pipeline [misc] refactor launch API and tensor constructor (#5666) 7 months ago
registry
tensor Fix ColoTensorSpec for py11 (#5440) 8 months ago
trainer
utils [feat] refactored extension module (#5298) 10 months ago
zero
__init__.py
constants.py
core.py
global_variables.py
initialize.py