pytorch/nn.attention.experimental.rst at 0da8127f77f9bf05ba204ea7659cb15ec85e88a7 - pytorch - Carlos Sousa's Git

OSSForks/pytorch

mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00

Boyuan Feng 68134a320e [Flex Attention] Paged Attention (#137164 )

This PR adds paged attention for flex attention.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/137164
Approved by: https://github.com/drisspg

2024-10-29 17:05:22 +00:00

8 lines

245 B

ReStructuredText

Raw Blame History

 torch.nn.attention.experimental
 ===============================
 .. currentmodule:: torch.nn.attention.experimental
 .. py:module:: torch.nn.attention.experimental
 .. warning::
    These APIs are experimental and subject to change without notice.