|
layers
|
Run isort and black on test files
|
2023-08-18 20:59:35 -07:00 |
|
losses
|
Run isort and black on test files
|
2023-08-18 20:59:35 -07:00 |
|
models
|
[Rotary] Implement rotary in Triton
|
2023-09-03 02:51:58 -07:00 |
|
modules
|
Run isort and black on test files
|
2023-08-18 20:59:35 -07:00 |
|
ops
|
Run isort and black on test files
|
2023-08-18 20:59:35 -07:00 |
|
test_flash_attn.py
|
Implement splitKV attention
|
2023-08-29 00:58:29 -07:00 |
|
test_rotary.py
|
[Rotary] Implement rotary in Triton
|
2023-09-03 02:51:58 -07:00 |
