Install
Docs
Pricing
Enterprise
More...
More...
Try Live
Rankings
Blog
Add Docs
Flash Attention Jax
https://github.com/lucidrains/flash-attention-jax
Admin
Flash Attention Jax is an implementation of the Flash Attention algorithm in Jax, providing
...
Tokens:
3,706
Snippets:
32
Trust Score:
9.9
License:
MIT
Update:
1 month ago
Context
Chat
Benchmark
Latest
Show doc for...
Code
Info
Show Results
Tokens:
Raw
Copy
Link