Ryu1845 / hyena-jaxLinks
Implementation of Hyena Hierarchy in JAX
☆10Updated 2 years ago
Alternatives and similar repositories for hyena-jax
Users that are interested in hyena-jax are comparing it to the libraries listed below
Sorting:
- FlexAttention w/ FlashAttention3 Support☆27Updated last year
- Triton Implementation of HyperAttention Algorithm☆48Updated last year
- Using FlexAttention to compute attention with different masking patterns