memory-efficient-attention-pytorch

Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"

artificial-intelligence

View on GitHub

394 Stars

37 Forks

7 Watchers

Python Language

mit License

100 SrcLog Score

Cost to Build

$2.17M

Market Value

$4.28M

How is this calculated?

Growth over time

7 data points · 2022-06-01 → 2026-08-01

Stars Forks Watchers

💬

How do you feel about this project?

Ask AI about memory-efficient-attention-pytorch

Question copied to clipboard

What is the lucidrains/memory-efficient-attention-pytorch GitHub project? Description: "Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"". Written in Python. Explain what it does, its main use cases, key features, and who would benefit from using it.

Question is copied to clipboard — paste it after the AI opens.

How to clone memory-efficient-attention-pytorch

Clone via HTTPS

git clone https://github.com/lucidrains/memory-efficient-attention-pytorch.git

Clone via SSH

[email protected]:lucidrains/memory-efficient-attention-pytorch.git

Download ZIP

Download main.zip

Found an issue?

Report bugs or request features on the memory-efficient-attention-pytorch issue tracker:

Open GitHub Issues

Similar to memory-efficient-attention-pytorch

machine-learning-for-software-engineers incubator-mxnet spaCy cheatsheets-ai deeplearning4j gun caffe2 warriorjs php-ml nupic-legacy AirSim Swift-AI SerpentAI ncnn TensorLayer reinforcement-learning-an-introduction EasyPR awesome-artificial-intelligence serenata-de-amor AlgoWiki pipeline mycroft-core IAMDinosaur shogun deep-learning-book gorgonia fairseq ELF carla nd4j