ring-attention-pytorch

Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch

MIT License

Downloads
5.4K
Stars
457
Committers
1
View Code on GitHub
Ecosystems: Python
ring-attention-pytorch - 0.2.4

Published by lucidrains 8 months ago

Full Changelog: https://github.com/lucidrains/ring-attention-pytorch/compare/0.2.3...0.2.4

ring-attention-pytorch - 0.2.3

Published by lucidrains 8 months ago

Full Changelog: https://github.com/lucidrains/ring-attention-pytorch/compare/0.2.2...0.2.3

ring-attention-pytorch - 0.2.2

Published by lucidrains 8 months ago

Full Changelog: https://github.com/lucidrains/ring-attention-pytorch/compare/0.2.1...0.2.2

ring-attention-pytorch - 0.2.1

Published by lucidrains 8 months ago

Full Changelog: https://github.com/lucidrains/ring-attention-pytorch/compare/0.2.0...0.2.1

ring-attention-pytorch - 0.2.0

Published by lucidrains 8 months ago

Full Changelog: https://github.com/lucidrains/ring-attention-pytorch/compare/0.1.48...0.2.0

ring-attention-pytorch - 0.1.48

Published by lucidrains 8 months ago

Full Changelog: https://github.com/lucidrains/ring-attention-pytorch/compare/0.1.47...0.1.48

ring-attention-pytorch - 0.1.47

Published by lucidrains 8 months ago

Full Changelog: https://github.com/lucidrains/ring-attention-pytorch/compare/0.1.46...0.1.47

ring-attention-pytorch - 0.1.46

Published by lucidrains 8 months ago

Full Changelog: https://github.com/lucidrains/ring-attention-pytorch/compare/0.1.45...0.1.46

ring-attention-pytorch - 0.1.45

Published by lucidrains 8 months ago

Full Changelog: https://github.com/lucidrains/ring-attention-pytorch/compare/0.1.44...0.1.45

ring-attention-pytorch - 0.1.44

Published by lucidrains 8 months ago

Full Changelog: https://github.com/lucidrains/ring-attention-pytorch/compare/0.1.43...0.1.44

ring-attention-pytorch - 0.1.43

Published by lucidrains 8 months ago

Full Changelog: https://github.com/lucidrains/ring-attention-pytorch/compare/0.1.42...0.1.43

ring-attention-pytorch - 0.1.42

Published by lucidrains 8 months ago

Full Changelog: https://github.com/lucidrains/ring-attention-pytorch/compare/0.1.41...0.1.42

ring-attention-pytorch - 0.1.41

Published by lucidrains 8 months ago

Full Changelog: https://github.com/lucidrains/ring-attention-pytorch/compare/0.1.40...0.1.41

ring-attention-pytorch - 0.1.40

Published by lucidrains 8 months ago

Full Changelog: https://github.com/lucidrains/ring-attention-pytorch/compare/0.1.39...0.1.40

ring-attention-pytorch - 0.1.39

Published by lucidrains 8 months ago

Full Changelog: https://github.com/lucidrains/ring-attention-pytorch/compare/0.1.37...0.1.39

ring-attention-pytorch - 0.1.38

Published by lucidrains 8 months ago

Full Changelog: https://github.com/lucidrains/ring-attention-pytorch/compare/0.1.37...0.1.38

ring-attention-pytorch - 0.1.37

Published by lucidrains 8 months ago

Full Changelog: https://github.com/lucidrains/ring-attention-pytorch/compare/0.1.36...0.1.37

ring-attention-pytorch - 0.1.36

Published by lucidrains 8 months ago

Full Changelog: https://github.com/lucidrains/ring-attention-pytorch/compare/0.1.35...0.1.36

ring-attention-pytorch - 0.1.35

Published by lucidrains 8 months ago

Full Changelog: https://github.com/lucidrains/ring-attention-pytorch/compare/0.1.34...0.1.35

ring-attention-pytorch - 0.1.34

Published by lucidrains 8 months ago

Full Changelog: https://github.com/lucidrains/ring-attention-pytorch/compare/0.1.33...0.1.34

Package Rankings
Top 37.37% on Pypi.org
Related Projects
local-attention

An implementation of local windowed attention for language modeling

05 Jul 2020 375
equiformer-pytorch

Implementation of the Equiformer, SE3/E3 equivariant attention network that reaches new SOTA, and...

29 Oct 2022 242
nuwa-pytorch

Implementation of NÜWA, state of the art attention network for text to video synthesis, in Pytorch

28 Nov 2021 540
soundstorm-pytorch

Implementation of SoundStorm, Efficient Parallel Audio Generation from Google Deepmind, in Pytorch

17 May 2023 1,370
phenaki-pytorch

Implementation of Phenaki Video, which uses Mask GIT to produce text guided videos of up to 2 min...

29 Sep 2022 747
CoLT5-attention

Implementation of the conditionally routed attention in the CoLT5 architecture, in Pytorch

20 Mar 2023 223
iTransformer

Unofficial implementation of iTransformer - SOTA Time Series Forecasting using Attention networks...

11 Oct 2023 429
audiolm-pytorch

Implementation of AudioLM, a SOTA Language Modeling Approach to Audio Generation out of Google Re...

09 Sep 2022 2,399
PaLM-rlhf-pytorch

Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the PaLM architectu...

09 Dec 2022 7,595
En-transformer

Implementation of E(n)-Transformer, which incorporates attention mechanisms into Welling's E(n)-E...

27 Feb 2021 208
meshgpt-pytorch

Implementation of MeshGPT, SOTA Mesh generation using Attention, in Pytorch

29 Nov 2023 642
magvit2-pytorch

Implementation of MagViT2 Tokenizer in Pytorch

10 Oct 2023 552
flash-attention

Fast and memory-efficient exact attention

19 May 2022 11,791
simple-hierarchical-transformer

Experiments around a simple idea for inducing multiple hierarchical predictive model within a GPT

06 Apr 2023 204
performer-pytorch

An implementation of Performer, a linear attention-based transformer, in Pytorch

03 Oct 2020 1,084