{"library":"e3nn-jax","install":[{"cmd":"pip install e3nn-jax","imports":["from e3nn_jax import Irreps","from e3nn_jax import spherical_harmonics","from e3nn_jax import rand_irreps","from e3nn_jax.flax import Linear"]},{"cmd":"pip install --upgrade \"jax[cuda12_pip]\" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html\npip install e3nn-jax","imports":[]}]}