/jax-geometric

Implementation of various equivariant models in JAX

Primary LanguagePythonMIT LicenseMIT

Stargazers