PyPI page
Home page
Author:
None
Summary:
JAX-Library for building E(3)-equivariant deep learning architectures based on Flax.
Latest version:
1.0.2
Required dependencies:
absl-py
|
etils
|
flax
|
jax
|
jaxtyping
|
more_itertools
|
numpy
|
sympy
Optional dependencies:
ipykernel
|
jupyter_sphinx
|
matplotlib
|
nbsphinx
|
pylint
|
pypandoc_binary
|
pytest
|
pytest-xdist
|
sphinx
|
sphinx-autodoc-typehints
|
sphinx-rtd-theme
|
yapf
Downloads last day:
73
Downloads last week:
374
Downloads last month:
1,701