https://github.com/graphcore-research/flash-attention-ipu
Poplar implementation of FlashAttention for IPU
https://github.com/graphcore-research/flash-attention-ipu
deep-learning flash-attention flash-attention-2 graphcore ipu poplar pytorch transformers
Last synced: 5 months ago
JSON representation
Poplar implementation of FlashAttention for IPU
- Host: GitHub
- URL: https://github.com/graphcore-research/flash-attention-ipu
- Owner: graphcore-research
- License: mit
- Created: 2023-07-25T13:44:22.000Z (almost 3 years ago)
- Default Branch: main
- Last Pushed: 2024-03-12T11:39:49.000Z (over 2 years ago)
- Last Synced: 2024-03-12T13:02:32.844Z (over 2 years ago)
- Topics: deep-learning, flash-attention, flash-attention-2, graphcore, ipu, poplar, pytorch, transformers
- Language: C++
- Homepage:
- Size: 575 KB
- Stars: 2
- Watchers: 4
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE