Code for SIGGRAPH paper CNNs on Surfaces using Rotation-Equivariant Features

Non-parametric hypothesis tests for identifying distributional group symmetries from data

Exploring the design surface of ACE for crystal deep learning

Script to generate STL of spherical harmonics

Compute Lyapunov exponents and Covariant-Lyapunov-Vectors of an RNN update trajectory

Steerable convolutions

ImageNet1k-pretrained SE(2) Equivariant Vision Models

Code for "Effect of equivariance on training dynamics"

Original code for "Exploiting Learned Symmetries in Group Equivariant Convolutions"

Experiments with Group Equivariant Convolutional Networks

Implementation of Group-Convolutions as Keras layers.

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

Official source code for "Latent Field Discovery in Interacting Dynamical Systems with Neural Fields". In NeurIPS 2023.

Randomization Tests for Conditional Group Symmetry

Image to Icosahedral Projection for SO(3) Object Reasoning from Single-View Images

Practical Equivariances via Relational Conditional Neural Processes (Huang et al., NeurIPS 2023)

Implementation of VectorNeuron-Transformer paper

Steerable E(3) GNN in jax

Official code for NeurIPS 2024 paper "A Canonicalization Perspective on Invariant and Equivariant Learning".

Lorentz-equivariant Transformer based on Lorentz Local Canonicalization (Spinner et al. 2025)