#

equivariant-neural-networks

Here are 10 public repositories matching this topic...

atomicarchitects / equiformer_v2

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

machine-learning deep-learning molecular-dynamics pytorch computational-chemistry computational-physics drug-discovery materials-science force-fields geometric-deep-learning graph-neural-networks interatomic-potentials ai-for-science e3nn equivariant-neural-networks equivariant-graph-neural-network catalyst-design

Updated Feb 11, 2025
Python

atomicarchitects / equiformer

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

machine-learning deep-learning molecular-dynamics pytorch computational-chemistry drug-discovery materials-science force-fields geometric-deep-learning graph-neural-networks interatomic-potentials ai-for-science e3nn equivariant-neural-networks equivariant-graph-neural-network catalyst-design

Updated Feb 11, 2025
Python

MorphoSymm

Danfoa / MorphoSymm

Tools for exploiting Morphological Symmetries in robotics

robotics dynamical-systems representation-learning group-theory representation-theory symmetry-analysis legged-robotics symmetries invariant-learning equivariant-control symmetric-robots equivariant-neural-networks morphological-symmetries

Updated Jan 8, 2026
Python

arnab39 / equiadapt

Library to make any existing neural network architecture equivariant

computer-vision deep-learning image-classification image-segmentation nbody-simulation geometric-deep-learning equivariance pointcloud-segmentation pointcloud-processing equivariant-neural-networks

Updated Nov 3, 2024
Jupyter Notebook

Infatoshi / batmobile

High-performance CUDA kernels for equivariant graph neural networks (MACE, NequIP, Allegro). 10-20x faster than e3nn.

machine-learning cuda molecular-dynamics pytorch gpu-computing mace e3nn equivariant-neural-networks

Updated Jan 17, 2026
Cuda

smsharma / eqnn-jax

Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.

symmetry n-body-simulation egnn equivariant-neural-networks segnn nequip

Updated Mar 1, 2025
Python

brian-hepler-phd / Spherical-CNN

Interactive exploration of equivariant neural networks on homogeneous spaces, with a focus on the sphere S² as SO(3)/SO(2). From Lecture 8 of the Lie groups course with Quantum Formalism

machine-learning pytorch scientific-computing lie-groups representation-theory geometric-deep-learning spherical-cnn homogeneous-spaces e3nn equivariant-neural-networks

Updated May 6, 2025
Jupyter Notebook

VITA-Group / TAPE

[ICML'25] "Rethinking Addressing in Language Models via Contextualized Equivariant Positional Encoding" by Jiajun Zhu, Peihao Wang, Ruisi Cai, Jason D. Lee, Pan Li, Zhangyang Wang

positional-encoding large-language-models equivariant-neural-networks

Updated Jun 6, 2025
Python

Danfoa / symmetric_learning

Torch modules for equivariant/invariant learning using ESCNN

deep-learning symmetry geometric-deep-learning equivariant-network equivariant-neural-networks

Updated Jan 13, 2026
Python

Joeshmoe03 / EquivariantHistopathology

A convolution-equivariant deep-learning approach to segmentation of histopathology data

deep-learning pytorch medical-imaging ocelot segmentation unet histopathology equivariance equivariant-neural-networks

Updated May 19, 2025
Jupyter Notebook

Improve this page

Add a description, image, and links to the equivariant-neural-networks topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the equivariant-neural-networks topic, visit your repo's landing page and select "manage topics."