pytc.jastrow.test.test_poly

Tests for JAX SimpleJastrow implementation.

Classes

TestSimpleJastrowJAX

Test cases for SimpleJastrowJAX class.

Functions

numerical_gradient_params	Compute numerical gradient with respect to parameters for single points.
numerical_gradient_r1	Compute numerical gradient with respect to r1 for single points.