We release the methods we develop as open-source software, so that others can use, check and build on them.
T
tensorialMachine learning on physical tensors: equivariant graph neural networks and machine-learned interatomic potentials in JAX.
JAXe3nn-jaxEquivariant NNsMLIPs