Jax EFT forward model

• Differentiable implementation of the field-level forward model based on EFT
• Jax version of Hi-Fi mocks

Check out this notebook on how to perform 3D field-level inference using NUTS HMC with NumPyro, varying initial conditions, cosmology and bias parameters.

Acknowledgement

This code evolved from JaxPM.