Popular GitHub Repositories for Shift Equivariant

pip install antialiased-cnns to improve stability and accuracy

Lightweight, useful implementation of conformal prediction on real data.

Geometric GNN Dojo provides unified implementations and experiments to explore the design space of Geometric Graph Neural Networks (ICML 2023)

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

Training neural network potentials

Official code of the paper "ReDet: A Rotation-Equivariant Detector for Aerial Object Detection" (CVPR 2021)

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

[ICLR 2024] EquiformerV2: Improved Equivariant Transformer for Scaling to Higher-Degree Representations

Implementation of the Equiformer, SE3/E3 equivariant attention network that reaches new SOTA, and adopted for use by EquiFold for protein folding

[ICLR 2023 Spotlight] Equiformer: Equivariant Graph Attention Transformer for 3D Atomistic Graphs

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

Frouros: an open-source Python library for drift detection in machine learning systems.

[ECCV 2022] Official PyTorch Code of DEVIANT: Depth Equivariant Network for Monocular 3D Object Detection

jax library for E3 Equivariant Neural Networks

GOOD: A Graph Out-of-Distribution Benchmark [NeurIPS 2022 Datasets and Benchmarks]

Transformation-Equivariant 3D Object Detection for Autonomous Driving

Pitch-shift audio clips quickly with PyTorch (CUDA supported)! Additional utilities for searching efficient transformations are included.

A short and easy PyTorch implementation of E(n) Equivariant Graph Neural Networks

[NeurIPS 2022] Learning Causally Invariant Representations for Out-of-Distribution Generalization on Graphs

List of Geometric GNNs for 3D atomic systems