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
PASSIONLab
openequivariance

OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.

184K 141 9
QUVA-Lab
e2cnn

E(2)-Equivariant CNNs Library for Pytorch

15K 674 84
lucidrains
en-transformer

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

6K 226 31
QUVA-Lab
escnn

Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/

5K 519 63
lucidrains
egnn-pytorch

Implementation of E(n)-Equivariant Graph Neural Networks, in Pytorch

4K 525 86
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
equiformer-pytorch

Equiformer - SE3/E3 Graph Attention Transformer for Molecules and Proteins

1K 289 31
lucidrains
lie-transformer-pytorch

Implementation of Lie Transformer, Equivariant Self-Attention, in Pytorch

1K 97 10
mfinzi
emlp

A library for programmatically generating equivariant layers through constraint solving

608 282 26
lucidrains
geometric-vector-perceptron

Geometric Vector Perceptron - Pytorch

594 77 13
PASSIONLab
openequivariance-extjax

OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.

541 141 9
atomicarchitects
harmonyx

Fully-Differentiable Tensor Spherical Harmonics in JAX

536 19 1
facundoq
tmeasures

The Transformational Measures (TM) library allows neural network researchers to evaluate the invariance and equivariance of their models with respect to a set of transformations. Support for Pytorch (current) and Tensorflow/Keras (coming).

368 3 1
chao1224
geom3d

Geom3D: Geometric Modeling on 3D Structures, NeurIPS 2023

145 130 15
arnab39
equiadapt

Library to make any existing neural network architecture equivariant

127 58 5
    • Data from PyPI, GitHub, ClickHouse, and BigQuery