PyPI page
Home page
Author:
None
Summary:
Machine Learning Interatomic Potentials in JAX
Latest version:
0.2.2
Required dependencies:
ase
|
dm-haiku
|
dm-tree
|
e3j
|
e3nn-jax
|
flax
|
h5py
|
jax
|
jax-md
|
matscipy
|
numpy
|
omegaconf
|
orbax-checkpoint
|
pydantic
|
roundmantissa
|
tqdm
|
tqdm-loggable
Optional dependencies:
e3j-ops
|
jax
|
jaxlib
|
libtpu
Downloads last day:
23
Downloads last week:
437
Downloads last month:
1,471