pytc.kmat.contract_K3_isdf_jit¶ pytc.kmat.contract_K3_isdf_jit(phi_p, phi_q, phi_r, phi_s, U3, rank_block_size=128)[source]¶ JITted version of K3 contraction. Parameters: rank_block_size – Block size for scanning the ISDF rank dimension. This is a static argument — JAX recompiles if it changes.