12 Forks
83 Stars
83 Watchers

flash-attention-jax

Implementation of Flash Attention in Jax

How to download and setup flash-attention-jax

Open terminal and run command
git clone https://github.com/lucidrains/flash-attention-jax.git
git clone is used to create a copy or clone of flash-attention-jax repositories. You pass git clone a repository URL.
it supports a few different network protocols and corresponding URL formats.

Also you may download zip file with flash-attention-jax https://github.com/lucidrains/flash-attention-jax/archive/master.zip

Or simply clone flash-attention-jax with SSH
[email protected]:lucidrains/flash-attention-jax.git

If you have some problems with flash-attention-jax

You may open issue on flash-attention-jax support forum (system) here: https://github.com/lucidrains/flash-attention-jax/issues