Equivariant Neural Networks Python Packages

avitai-artifex

Generative modeling for JAX/Flax NNX: VAEs, GANs, DDPM/DDIM/score/DiT/latent diffusion, normalizing flows, energy-based, autoregressive and geometric (SE(3), protein, point cloud, mesh) models; image, text, audio, molecular, protein, tabular and time-series modalities; a shared Trainer with callbacks and extensions; evaluation through calibrax.

flashcart

Fast Cartesian tensor operations and equivariant interatomic potentials for atomistic simulations

equiadapt

Library to make any existing neural network architecture equivariant

sigorbit

SigOrbit: turn rotated signatures into stable 256-D embeddings

morpho-symm

Tools for exploiting Morphological Symmetries in robotics