Error
Looks like something went wrong!

About

A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.

Apache-2.0 license

Security policy

Custom properties

Report repository

Releases 1

Packages

No packages published

Languages

Python 100.0%