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/nn
Edenzzzz f5c84af0b0
[Feature] Zigzag Ring attention (#5905)
3 months ago
..
_ops [misc] update pre-commit and run all files (#4752) 1 year ago
layer [Feature] Zigzag Ring attention (#5905) 3 months ago
loss [npu] change device to accelerator api (#5239) 11 months ago
metric [misc] update pre-commit and run all files (#4752) 1 year ago
parallel [misc] update pre-commit and run all files (#4752) 1 year ago
__init__.py