Search All packages Top packages Track packages
PyPI page Summary: FlashAttention for JAX: Dao-AILab's CUDA kernels as XLA FFI targets Latest version: 0.3.3 Required dependencies: jax Downloads last day: 103 Downloads last week: 287 Downloads last month: 0 7-day smoothing