pet-jax

PyPI page
Author: Marcel Langer
License: MIT
Summary: Clean JAX/Flax implementation of uPET interatomic potentials
Latest version: 0.1.0
Required dependencies: ase | flax | jax | jaxlib | marathon-train | numpy | vesin
Optional dependencies: metatomic-torch | metatrain | torch

Downloads last day: 2
Downloads last week: 8
Downloads last month: 105
7-day smoothing