jaxeffort

Repo containing the jaxeffort emulator.

Installation and usage

In order to install jaxeffort, clone this repository, enter it, and run

pip install .

In order to use the emulators, you have to import jaxeffort and load a trained emulator

import jaxeffort
import jax.numpy as np
trained_emu = jaxeffort.load_multipole_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
bias_array = np.array([...])
result = trained_emu.get_Pl(input_array, bias_array)

For a more detailed explanation, check the tutorial in the notebooks folder, which also shows a comparison with the standard `calculations.