I would like to see a Tensorflow local or TPU integration for LAMMPS

LAMMPS docs.lammps org/
Has a plugin called ML-IAP plugin for PyTorch. I would like to cloud on Windows 11 or Linux a TPU optimization for energy potentials in force fields for Molecular Dynamics analysis. LAMMPS is paid for by the U.S. Government Department of Energy and is free to use and Modify for U.S. Corporations and U.S. Citizens.

I am requesting a cloud package installation with the endpoint and an I/O.

I could build it myself but the Google BLAS-LAPACK optimizations for their TPU system since their version of BLAS-LAPACK is not open source, I cannot work on it.

Thanks,

Aaron Linder @ludiusvox (github)