Repo containing the jaxcapse emulator.
In order to install jaxcapse, clone this repository, enter it, and run
pip install .In order to use the emulators, you have to import jaxcapse and load a trained emulator
import jaxcapse
import jax.numpy as np
trained_emu = jaxcapse.load_emulator("/path/to/emu/")Then you are good to! You have to create an input array and retrieve your calculation result
input_array = np.array([...]) #write in the relevant numbers
result = trained_emu.get_Cl(input_array)For a more detailed explanation, check the tutorial in the notebooks folder, which also shows a comparison with the standard CAMB Boltzmann solver.
Free usage of the software in this repository is provided, given that you cite our release paper.
M. Bonici, F. Bianchini, J. Ruiz-Zapatero, Capse: efficient and auto-differentiable CMB power spectra emulation