flash-mla / README.md

drbh

feat: build flash mla with kernel builder

1f83cde 13 days ago

338 Bytes

metadata

tags:
  - kernel
  - flash-mla
  - deepseek
  - kernel-builder

flash-mla

This repo builds Deepseeks FlashMLA kernel via the HF kernel-builder

nix develop -L
pytest -vv tests/

nix build .#bundle -L