Equivariant machine learning interatomic potentials in JAX. - View it on GitHub