Teddy Koker
banner
teddykoker.com
Teddy Koker
@teddykoker.com
PhD student at MIT; AI for materials. teddykoker.com
With slight changes of the initialization and hyperparameters, we achieve improved RMSE on the 3BPA test set! This work serves as a simple introduction to training and testing equivariant neural network potentials in JAX, and can be easily extended for new methodologies.

n/n
May 22, 2025 at 3:40 PM
Introducing nequip-eqx, a JAX implementation of the popular NequIP interatomic potential model.

Repo: github.com/teddykoker/n...

The goal of the repository is to offer a simple (<1000 lines of code) implementation while providing competitive performance to existing codebases.

1/n
May 22, 2025 at 3:40 PM