An implementation of Tree-Attention in PyTorch because it's in JAX for some reason
python3 model.pyMIT
- Implement flash attention from the native official repo, I couldn't because the docs are nowhere to be found and understood
kyegomez/Tree-Attention-Torch
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|
Repository files navigation
Releases
No releases published
Sponsor this project
Generated from kyegomez/Python-Package-Template