File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change 1
1
# Memory Efficient Attention
2
2
[ ![ arXiv] ( https://img.shields.io/badge/arXiv-2112.05682v2-b31b1b.svg?style=flat-square )] ( https://arxiv.org/abs/2112.05682v2 )
3
- [ ![ PyPI version] ( https://img.shields.io/badge/memory--efficient--attention-0.1.2 -informational?style=flat-square&color=C51BA3 )] ( https://pypi.org/project/memory-efficient-attention/ )
3
+ [ ![ PyPI version] ( https://img.shields.io/badge/memory--efficient--attention-0.1.3 -informational?style=flat-square&color=C51BA3 )] ( https://pypi.org/project/memory-efficient-attention/ )
4
4
5
5
This is ** unofficial** implementation of [ Self-attention Does Not Need O(n^2) Memory] ( https://arxiv.org/abs/2112.05682v2 ) for Jax and PyTorch.
6
6
Original file line number Diff line number Diff line change 2
2
3
3
setup (
4
4
name = 'memory-efficient-attention' ,
5
- version = '0.1.2 ' ,
5
+ version = '0.1.3 ' ,
6
6
description = 'Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch' ,
7
7
license = 'MIT' ,
8
8
packages = find_packages (),
You can’t perform that action at this time.
0 commit comments