pytc.tc._compute_2b_shard

pytc.tc._compute_2b_shard(phi, grad_phi, grid, weights, jastrow_params, jastrow_factor, ranges, batch_size)[source]

Compute K terms for one device shard.