sample highly correlated bivariate normal based on jax, autograd, pytorch, or pure numpy
hmc for jax, autograd, pytorch, numpy