Introducing a universal NumPy interface for various NumPy-compatible libraries#
A general NumPy wrapper can be developed in PySCF to provide a universal interface
to access NumPy functions. This interface would enable automatic switching to
jax.numpy
or torch
functions depending on the runtime context.
This interface can be imported as
import pyscf.numpy as fnp