/toy_flash_attn

Demonstration implementation of FlashAttention using nbdev.

Primary LanguageJupyter NotebookApache License 2.0Apache-2.0

toy_flash_attn

This file will become your README and also the index of your documentation.

Install

pip install toy_flash_attn

How to use

Fill me in please! Don’t forget code examples:

1+1
2