Science Score: 54.0%
This score indicates how likely this project is to be science-related based on various indicators:
-
✓CITATION.cff file
Found CITATION.cff file -
✓codemeta.json file
Found codemeta.json file -
○.zenodo.json file
-
✓DOI references
Found 2 DOI reference(s) in README -
✓Academic publication links
Links to: arxiv.org -
○Academic email domains
-
○Institutional organization owner
-
○JOSS paper metadata
-
○Scientific vocabulary similarity
Low similarity (16.2%) to scientific vocabulary
Keywords
Repository
JAX-based linear Einstein-Boltzmann solver for cosmology
Basic Info
- Host: GitHub
- Owner: ohahn
- License: gpl-3.0
- Language: Jupyter Notebook
- Default Branch: master
- Homepage: https://doi.org/10.1088/1475-7516/2024/06/063
- Size: 41.7 MB
Statistics
- Stars: 19
- Watchers: 4
- Forks: 6
- Open Issues: 1
- Releases: 1
Topics
Metadata Files
README.md
DISCO-EB - The DISCO-DJ Einstein-Boltzmann module
Implementation of a differentiable linear Einstein-Boltzmann solver for cosmology in JAX -- a module of the DISCO-DJ framework (DIfferentiable Simulations for COsmology, Done with Jax).
Note that this program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY.
Currently supported features (the list is growing, so check back):
- Autodifferentiable via JAX
- Standard LCDM model with Quintessence DE fluid (w0,wa) and massive neutrinos (one species)
- Thermal history solver based on a simplified Recfast implementation in the module
- Numerous example jupyter notebooks, e.g. for Euclid-like Fisher forecasts
New modules/plugins can be easily added (see how to contribute in CONTRIBUTING.md file). We are enthusiastic if extensions/improvements that are of broader interest are re-integrated into the master branch and DISCO-EB grows as a community effort.
Currently work in progress:
- performance improvements
Installation
Simple Install
bash
pip install git+https://github.com/ohahn/DISCO-EB.git
For Development
In a fresh virtual environment:
bash
git clone https://github.com/ohahn/DISCO-EB.git
cd DISCO-EB
pip install -e .
Running the tests
First install the additional dependencies with the [dev] flag:
bash
cd DISCO-EB
pip install -e '.[dev]'
Then run pytest to start the tests. This should be run on GPU platforms.
Getting started
Start with the sample notebook nbminimalexample.ipynb in the notebooks subdirectory. It explains how to compute a matter power spectrum and take a derivative w.r.t. a cosmological parameter.
Benchmarks
See BENCHMARKS.md for benchmark results on different hardware.
Contributing and Licensing
See file CONTRIBUTING.md on how to contribute to the development.
The software is licensed under GPL v3 (see file LICENSE).
Citing in scientific publications or presentations
If you use DISCO-EB in your scientific work, you are required to acknowledge this by linking to this repository and citing the relevant papers:
- Hahn et al. 2024 arXiv:2311.03291 JCAP OA
Required Python Packages
- JAX
- diffrax==0.4.1
- equinox
- jaxtyping
- jax_cosmo
Owner
- Name: Oliver Hahn
- Login: ohahn
- Kind: user
- Repositories: 2
- Profile: https://github.com/ohahn
Citation (CITATION.bib)
@article{DISCO-EB:2024,
author = "Hahn, Oliver and List, Florian and Porqueres, Natalia",
title = "{DISCO-DJ I: a differentiable Einstein-Boltzmann solver for cosmology}",
eprint = "2311.03291",
archivePrefix = "arXiv",
primaryClass = "astro-ph.CO",
doi = "10.1088/1475-7516/2024/06/063",
journal = "JCAP",
volume = "06",
pages = "063",
year = "2024"
}
GitHub Events
Total
- Issues event: 1
- Watch event: 9
- Delete event: 3
- Issue comment event: 1
- Push event: 10
- Pull request event: 4
- Fork event: 3
- Create event: 4
Last Year
- Issues event: 1
- Watch event: 9
- Delete event: 3
- Issue comment event: 1
- Push event: 10
- Pull request event: 4
- Fork event: 3
- Create event: 4
Dependencies
- diffrax ==0.4.1
- equinox *
- jax *
- jax_cosmo *
- jaxtyping *