pyscf-ipu
pyscf-ipu copied to clipboard
Enable pmap for ERI value computation and sparse einsum and DFT integration
This PR enables pmap for the intor version of the sparse einsum and integrates them with the main DFT code