ELS-RD/kernl

Kernl lets you run PyTorch transformer models several times faster on GPU with a single line of code, and is designed to be easily hackable.

语言：Jupyter Notebook · GitHub 星标：1,584 · Fork：98

话题：cuda · cuda-kernel · pytorch · transformer · triton

在 GitHub 上查看原仓库