piwheels

segnn-jax

Steerable E(3) GNN in jax

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install segnn-jax

Dependencies

• jax
• dm-haiku
• numpy
• e3nn-jax
• optax
• jraph
• jaxlib

Releases

Version	Released	Bullseye Python 3.9	Bookworm Python 3.11	Trixie Python 3.13	Files
0.7	2023-07-16
segnn_jax-0.7-py3-none-any.whl (12 KB)
0.6	2023-05-08
segnn_jax-0.6-py3-none-any.whl (11 KB)
0.5	2023-03-31
segnn_jax-0.5-py3-none-any.whl (11 KB)
0.3	2023-01-20
segnn_jax-0.3-py3-none-any.whl (9 KB)
0.2	2022-12-15
segnn_jax-0.2-py3-none-any.whl (9 KB)

Issues with this package?

• Search issues for this package
• Package or version missing? Open a new issue
• Something else? Open a new issue