PyPI Stats
  • Insights
  • PyPI
  • GitHub
  • Search
  • Compare
  • Advisories
  • Ecosystem
  • About
Home

Search Packages

Find Python packages by name, description, GitHub topic, or filter by metrics
lucidrains
x-transformers

A concise but complete full-attention transformer with a set of promising experimental features from various papers

1.4M 6K 508
lucidrains
vit-pytorch

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

109K 25K 3K
lucidrains
linear-attention-transformer

Transformer based on a variant of attention that is linear complexity in respect to sequence length

73K 829 76
lucidrains
linformer

Implementation of Linformer for Pytorch

72K 306 27
linto-ai
whisper-timestamped

Multilingual Automatic Speech Recognition with word-level timestamps and confidence

49K 3K 210
lucidrains
reformer-pytorch

Reformer, the efficient Transformer, in Pytorch

26K 2K 252
lucidrains
flamingo-pytorch

Implementation of 🦩 Flamingo, state-of-the-art few-shot visual question answering attention net out of Deepmind, in Pytorch

21K 1K 66
kyegomez
zetascale

Build high-performance AI models with modular building blocks

19K 590 57
lucidrains
tab-transformer-pytorch

Implementation of TabTransformer, attention network for tabular data, in Pytorch

13K 1K 131
lucidrains
enformer-pytorch

Implementation of Enformer, Deepmind's attention network for predicting gene expression, in Pytorch

12K 564 101
CyberZHG
keras-self-attention

Attention mechanism for processing sequential data that considers the context for each timestamp.

10K 657 117
lucidrains
performer-pytorch

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

10K 1K 150
lucidrains
alphafold2-pytorch

To eventually become an unofficial Pytorch implementation / replication of Alphafold2, as details of the architecture get released

8K 2K 266
lucidrains
en-transformer

Implementation of E(n)-Transformer, which incorporates attention mechanisms into Welling's E(n)-Equivariant Graph Neural Network

6K 226 31
lucidrains
dalle-pytorch

Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch

6K 6K 643
lucidrains
ring-attention-pytorch

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

5K 548 35
densechen
activations

AReLU: Attention-based-Rectified-Linear-Unit

5K 62 8
lucidrains
x-transformers-rl

Implementation of a transformer for reinforcement learning using `x-transformers`

4K 73 6
lucidrains
se3-transformer-pytorch

Implementation of SE3-Transformers for Equivariant Self-Attention, in Pytorch. This specific repository is geared towards integration with eventual Alphafold2 replication.

4K 327 30
lucidrains
perceiver-pytorch

Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch

4K 1K 138
awslabs
sockeye

Sequence-to-sequence framework with a focus on Neural Machine Translation based on PyTorch

3K 1K 320
lucidrains
nystrom-attention

Implementation of Nyström Self-attention, from the paper Nyströmformer

3K 145 18
lucidrains
axial-attention

Implementation of Axial attention - attending to multi-dimensional data efficiently

3K 396 32
lucidrains
nuwa-pytorch

NÜWA - Pytorch

3K 549 56
    • Data from PyPI, GitHub, ClickHouse, and BigQuery