e3nn/e3nn-jax

jax library for E3 Equivariant Neural Networks

/ 100

Established

224 stars and 158,867 monthly downloads. Used by 2 other packages. No commits in the last 6 months. Available on PyPI.

Stale 6m

Maintenance 2 / 25

Adoption 22 / 25

Maturity 25 / 25

Community 13 / 25

Stars

224

Forks

Language

Python

License

Apache-2.0

Category

Last pushed

Aug 25, 2025

Monthly downloads

158,867

Commits (30d)

Dependencies

Reverse dependents

Get this data via API

curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/e3nn/e3nn-jax"

Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.

Compare

e3nn-jax and equinox

google-deepmind/optax

Optax is a gradient processing and optimization library for JAX.

google/grain

Library for reading and processing ML training data.

patrick-kidger/equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

explosion/thinc

🔮 A refreshing functional take on deep learning, compatible with your favorite libraries

extropic-ai/thrml

Thermodynamic Hypergraphical Model Library in JAX