einops

Depended on by

• ABlooper

• adjacent-attention-pytorch

• alphafold2

• alphafold2-pytorch

• anymal-belief-state-encoder-decoder-pytorch

• augmax

• bidirectional-cross-attention

• big-sleep

• bottleneck-transformer-flax

• bottleneck-transformer-pytorch

• carefree-learn

• charformer-pytorch

• chili-etc

• CHILI-ETC

• CoCa-pytorch

• coco-lm-pytorch

• compositional-attention

• compositional-attention-pytorch

• comvex

• conformer

• conformer-tf

• cross-transformers-pytorch

• dalle2-pytorch

• dalle-pytorch

• dalle-pytorch-dev

• ddpm-proteins

• deep-daze

• deepvit-flax

• deformable-attention

• denoising-diffusion-pytorch

• dpfp-pytorch

• egnn-pytorch

• einops-exts

• enformer-pytorch

• En-transformer

• esbn-pytorch

• ETSformer-pytorch

• fast-transformer

• fast-transformer-pytorch

• feedback-transformer-pytorch

• flamingo-pytorch

• FLASH-pytorch

• flax-extra

• gated-state-spaces-pytorch

• glom-pytorch

• glom-tf

• gmlp-flax

• g-mlp-gpt

• g-mlp-pytorch

• graph-transformer-pytorch

• gsa-pytorch

• halonet-pytorch

• hamburger-pytorch

• hourglass-transformer-pytorch

• htm-pytorch

• h-transformer-1d

• imagen-pytorch

• invariant-point-attention

• isab-pytorch

• ITTR-pytorch

• jam-dist

• keras-cv-attention-models

• knowledge-neurons

• kronecker-attention-pytorch

• lambdanet-tf

• lambda-networks

• lie-transformer-pytorch

• lightweight-gan

• lit-saint

• local-attention-flax

• logavgexp-pytorch

• long-short-transformer

• marcopolo-pytorch

• marge-pytorch

• memformer

• memorizing-transformers-pytorch

• memory-efficient-attention-pytorch

• metaformer-gpt

• metnet

• mlp-gpt-jax

• mlpmixer-flax

• mlp-mixer-pytorch

• molecule-attention-transformer

• mp-nerf

• multistream-transformers

• n-grammer-pytorch

• nuwa-pytorch

• nwt-pytorch

• nystrom-attention

• omninet-pytorch

• opencoodx

• PaLM-jax

• PaLM-pytorch

• parti-pytorch

• perceiver

• perceiver-ar-pytorch

• perceiverio

• perceiver-io-pytorch

• perceiver-jax

• perceiver-pytorch

• performer-pytorch

• pi-gan-pytorch

• pixel-level-contrastive-learning

• point-transformer-pytorch

• ponder-transformer

• poolformer

• progen-transformer

• protein-bert-pytorch

• protein-glm

• pyglom

• rela-transformer

• remixer-pytorch

• res-mlp-pytorch

• retro-pytorch

• rotary-embedding-torch

• RQ-transformer

• se3-transformer-pytorch

• segformer-pytorch

• siren-pytorch

• solo-learn

• stam-pytorch

• tab-transformer-pytorch

• tf-bind-transformer

• tf-x-transformers

• tiggy

• tikt-performer-pytorch

• timesformer-pytorch

• tnt-tensorflow

• token-shift-gpt

• torcharc

• torchglyph

• torchrua

• tranception-pytorch

• transformer-in-transformer

• transformer-in-transformer-flax

• transganformer

• triangle-multiplicative-module

• triton-transformer

• tr-rosetta-pytorch

• uformer-pytorch

• uniformer-pytorch

• Unseal

• video-diffusion-pytorch

• vit-pytorch

• x-clip

• x-transformers

• x-unet