LICENSE
README.md
pyproject.toml
setup.cfg
setup.py
segnn_jax/__init__.py
segnn_jax/blocks.py
segnn_jax/graph_utils.py
segnn_jax/irreps_computer.py
segnn_jax/segnn.py
segnn_jax.egg-info/PKG-INFO
segnn_jax.egg-info/SOURCES.txt
segnn_jax.egg-info/dependency_links.txt
segnn_jax.egg-info/requires.txt
segnn_jax.egg-info/top_level.txt