Jingya
/

custom-nki-kernels

Model card Files Files and versions

custom-nki-kernels / src

94.9 kB

Ctrl+K

Ctrl+K

1 contributor

History: 8 commits

Jingya's picture

Jingya HF Staff

Upload src/pipeline.py with huggingface_hub

d38fb47 verified about 1 month ago

benchmark.py

19.2 kB
Upload src/benchmark.py with huggingface_hub about 1 month ago
nki_flash_attn.py

13.6 kB
Upload src/nki_flash_attn.py with huggingface_hub about 1 month ago
nki_qkv.py

13.9 kB
Upload src/nki_qkv.py with huggingface_hub about 1 month ago
pipeline.py

48.2 kB
Upload src/pipeline.py with huggingface_hub about 1 month ago