NX-AI/mlstm_kernels

Tiled Flash Linear Attention library for fast and efficient mLSTM Kernels.

语言：Jupyter Notebook · GitHub 星标：96 · Fork：9

话题：deep-learning · llm · rnn · triton-kernels · xlstm

在 GitHub 上查看原仓库