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.
 
 
 
 
 
Hanks 6991819a97
Merge branch 'hpcaitech:main' into feature/fp8_comm
5 months ago
..
_C
_analyzer
accelerator
amp
auto_parallel [pre-commit.ci] pre-commit autoupdate (#5572) 5 months ago
autochunk
booster [shardformer] fix the moe (#5883) 5 months ago
checkpoint_io [MoE/ZeRO] Moe refactor with zero refactor (#5821) 5 months ago
cli
cluster [MoE/ZeRO] Moe refactor with zero refactor (#5821) 5 months ago
context
device
fx
inference [pre-commit.ci] pre-commit autoupdate (#5572) 5 months ago
interface
kernel
lazy
legacy [pre-commit.ci] pre-commit autoupdate (#5572) 5 months ago
logging
moe [MoE/ZeRO] Moe refactor with zero refactor (#5821) 5 months ago
nn
pipeline
quantization Merge branch 'hpcaitech:main' into feature/fp8_comm 5 months ago
shardformer [shardformer] fix the moe (#5883) 5 months ago
tensor [MoE/ZeRO] Moe refactor with zero refactor (#5821) 5 months ago
testing
utils
zero [hotfix] fix the bug that large tensor exceed the maximum capacity of TensorBucket (#5879) 5 months ago
__init__.py
initialize.py