This repository implements a collection of Flash-Linear-Attention models that extend beyond language, supporting vision, video, and more.
-
$\texttt{[2025-01-25]}$ : This repo is created with vision models.
vision:fla-zoocurrently supports vision models. A simple documentation is in here.
Requirements:
- All the dependencies shown here
- torchvision
As an actively developed repo, no released packages are provided.
- Write documentation for vision models.
- Release training scripts for vision models.