examples/sparse_fock.py demonstrates screening in isolation. The realistic downstream step is contracting the screened integrals against a density matrix via GEMM — which is trnblas.
Acceptance: extend the demo to build a screened ERI tensor, then compute F = contract(screened_ERI, D) via trnblas.gemm / batched_gemm. Mirror the integration pattern from examples/quantum_chemistry/df_mp2_synthetic.py in the umbrella repo.
examples/sparse_fock.pydemonstrates screening in isolation. The realistic downstream step is contracting the screened integrals against a density matrix via GEMM — which is trnblas.Acceptance: extend the demo to build a screened ERI tensor, then compute
F = contract(screened_ERI, D)viatrnblas.gemm/batched_gemm. Mirror the integration pattern fromexamples/quantum_chemistry/df_mp2_synthetic.pyin the umbrella repo.