Giters

featherware / toy_flash_attn

Demonstration implementation of FlashAttention using nbdev.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About

Demonstration implementation of FlashAttention using nbdev.

License:Apache License 2.0

Languages

Language:Jupyter Notebook 59.1%Language:Python 34.9%Language:CSS 6.1%