JAX bindings for the flash-attention3 kernels
☆23Sep 9, 2026Updated last month
Alternatives and similar repositories for jax-flash-attn3
Users that are interested in jax-flash-attn3 are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- A ROS 1/ROS 2 hybrid package wrapping the Apache TVM project.☆10Jan 24, 2023Updated 3 years ago
- Llama-3.2-Vision study☆11Oct 11, 2024Updated 2 years ago