Apax: A Flexible and Performant Framework For The Development of Machine-Learned Interatomic Potentials

Moritz René Schäfer, Nico Segreto, Fabian Zills +2

We introduce Atomistic learned potentials in JAX (apax), a flexible and efficient open source software package for training and inference of machine-learned interatomic potentials.…