# Installation Instructions
To install the dependencies, run the following command:
pip install -U "jax[cuda12]" && \
pip install equinox matplotlib optax && \
pip install --force-reinstall "numpy<2" && \
pip install jax-cosmo && \
pip install 'numpyro[cuda]' -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html && \
pip install git+https://github.com/deepmind/dm-haiku && \
pip install -U scikit-learn && \
pip install --upgrade tensorflow-probability && \
pip install tensorflow && \
pip install tf-keras && \
pip install git+https://github.com/DifferentiableUniverseInitiative/sbi_lens.git && \
pip install sbi

Finally, to install the package, run the following command:
pip install -e .

# Reference
- The code for approxml/cosmo.py is taken from,
https://github.com/DifferentiableUniverseInitiative/sbi_lens/blob/main/sbi_lens/simulator/LogNormal_field.py
and https://github.com/DifferentiableUniverseInitiative/sbi_lens/blob/main/sbi_lens/simulator/redshift.py