E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.

E(3) Steerable Graph Neural Network

Paper list for equivariant neural network

Code repository for the paper "Group Equivariant Stand-Alone Self Attention For Vision" published at ICLR 2021. https://openreview.net/forum?id=JkfYjnOEo6M

Large-scale ligand-based virtual screening for potential SARS-Cov-2 inhibitors using a deep neural network