learning/annotated_deep_learning_paper_implementations
1
0
Fork 0
mirror of https://github.com/labmlai/annotated_deep_learning_paper_implementations.git synced 2025-10-27 20:24:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

flash attention

Browse Source
This commit is contained in:
Varuna Jayasiri
2025-08-08 19:57:57 +05:30
parent 4752644737
commit 9262c57f18
8 changed files with 1212 additions and 937 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
setup.py
View File

@ -5,7 +5,7 @@ with open("readme.md", "r", encoding="utf-8") as f:
setuptools.setup(
name='labml-nn',
version='0.4.137',
version='0.5.0',
author="Varuna Jayasiri, Nipun Wijerathne",
author_email="vpjayasiri@gmail.com, hnipun@gmail.com",
description="🧑‍🏫 Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit), optimizers (adam, radam, adabelief), gans(dcgan, cyclegan, stylegan2), 🎮 reinforcement learning (ppo, dqn), capsnet, distillation, diffusion, etc. 🧠",
@ -20,7 +20,7 @@ setuptools.setup(
'labml_helpers', 'labml_helpers.*',
'test',
'test.*')),
install_requires=['labml==0.4.168',
install_requires=['labml',
'torch',
'torchtext',
'torchvision',