GitHub repository: mitkotak/e3nn-jax
Primary language: Python
Description: jax library for E3 Equivariant Neural Networks