Kvax

https://github.com/nebius/kvax

Kvax is a fast, Triton-based Flash Attention 2 implementation for JAX that provides high-performance...

Tokens:10,350
Snippets:35
Trust Score:8.3
License:Apache-2.0
Update:7 months ago
Tokens:
Raw