jaxrts.ee_localfieldcorrections_dornheim_2021

This file is the taken from https://github.com/ToDor90/LFC/blob/master/ESA_PARAMETRIZATION/ collecting both scripts in one, replacing numpy by jax.numpy and performing some qol changes.

See [Dornheim et al., 2021], and references therein.

Functions

Activation(x, a, b)
Ef(rs)
G_analytical(k_over_k_f, rs, theta)
G_fit_wrap_extended(y, alpha, beta, gamma, delta)
Groth_A(rs, t)
OnTop(rs, t)
alpha_extended(rs, a, b, c)
beta_extended(rs, a, b, c)
delta_extended(rs, a, b, c)
density(rs)
double(n, beta)
f_extended(t, a, b, c)
gamma_extended(rs, a, b, c)
inverse_temperature(rs, t)
qf(rs)
x_m(theta)