DOE OSTI · code-167961
JVP Flash Attention (jvp_flash_attention) v0.0.4
Abstract
A Flash Attention Triton kernel with support for second-order derivatives, such as Jacobian-Vector Products (JVPs) and Hessian-Vector Products (HVPs).
Keep this discovery
Explore connections, maps & timelines
Morehead, Alex [Lawrence Berkeley National Laboratory (LBNL), Berkeley, CA (United States)]. 2025-10-20. JVP Flash Attention (jvp_flash_attention) v0.0.4. https://doi.org/10.5281/zenodo.17050188
Cite the original work for its findings. Save a collection to share your selection of sources.