/JaxPM-Hi-Fi-mocks

JAX-powered Hi-Fi mocks

Primary LanguageJupyter NotebookMIT LicenseMIT

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.