| .. |
|
triton
|
8c20cfef49
[Rotary] Support qkv block layout from GQA
|
3 месяцев назад |
|
__init__.py
|
ece539abd6
Add __init__.py files to subdirectories for installation
|
2 лет назад |
|
activations.py
|
3557e0bb8f
[MLP] Implement SwiGLU with torch jiterator
|
1 год назад |
|
fused_dense.py
|
0cb595ad94
[bugfix] handle_x not define when using checkpoint_lvl = 2 (#502)
|
1 год назад |
|
layer_norm.py
|
f1a73d0740
Run isort and black on python files
|
1 год назад |
|
rms_norm.py
|
f1a73d0740
Run isort and black on python files
|
1 год назад |
Tri Dao
8c20cfef49
[Rotary] Support qkv block layout from GQA