pytc.ansatz.sj.compute_jastrow_log_value

pytc.ansatz.sj.compute_jastrow_log_value(sj, elec_coords, jastrow_params)[source]

Compute Jastrow factor in log space for numerical stability.

Assumes unbatched elec_coords with shape (n_electrons, 3). Use vmap for batched processing.

Parameters:

sj (flax.struct.dataclass)