Skip to content

repositories Search Results · repo:nshepperd/flash_attn_jax language:C++

Filter by

0 files
 (81 ms)

0 files

innshepperd/flash_attn_jax (press backspace or delete to remove)

JAX bindings for Flash Attention v2
  • C++
  • 89
  • Updated
    on Jul 15, 2024
Package icon

Sponsor open source projects you depend on

Contributors are working behind the scenes to make open source better for everyone—give them the help and recognition they deserve.Explore sponsorable projects
ProTip! 
Press the
/
key to activate the search input again and adjust your query.
Package icon

Sponsor open source projects you depend on

Contributors are working behind the scenes to make open source better for everyone—give them the help and recognition they deserve.Explore sponsorable projects
ProTip! 
Press the
/
key to activate the search input again and adjust your query.