flash-hog

PyPI page
Home page
Author: None
Summary: Kernels for higher order gradients of Flash Attention.
Latest version: 0.6.0
Required dependencies: chex | einops | jax

Downloads last day: 0
Downloads last week: 33
Downloads last month: 302