https://github.com/colmtalbot/bilby-numpyro
A Bilby interface to numpyro samplers
Science Score: 13.0%
This score indicates how likely this project is to be science-related based on various indicators:
-
○CITATION.cff file
-
✓codemeta.json file
Found codemeta.json file -
○.zenodo.json file
-
○DOI references
-
○Academic publication links
-
○Academic email domains
-
○Institutional organization owner
-
○JOSS paper metadata
-
○Scientific vocabulary similarity
Low similarity (4.6%) to scientific vocabulary
Repository
A Bilby interface to numpyro samplers
Basic Info
- Host: GitHub
- Owner: ColmTalbot
- License: mit
- Language: Python
- Default Branch: main
- Size: 35.2 KB
Statistics
- Stars: 1
- Watchers: 2
- Forks: 2
- Open Issues: 3
- Releases: 1
Metadata Files
README.md
bilby-numpyro
Sampler plugin for using numpyro samplers with Bilby
bilby-numpyro is a sampler plugin for using numpyro samplers with Bilby likelihoods and priors.
Any likelihood that uses only JAX JIT-compilable functions/classes works automatically.
Usage
The sampler should always be given as numpyro regardless of which sampler (e.g., HMC, NUTS) is actually being used.
The sampler is specified with sampler_name and any sampler keyword arguments should be passed in the usual way to
bilby.core.sampler.run_sampler for example
python
run_sampler(
...,
sampler="numpyro",
sampler_name="NUTS",
num_warmup=500,
num_samples=1000,
)
This has not been tested with numpyro parallelisation.
Limitations
The likelihood class you use should be JIT compilable with JAX.
Generally, this will be the case if you use jax.numpy and jax.scipy instead of numpy/scipy.
Some Bilby priors have direct equivalents in numpyro and so we use those automatically.
You will see logged warnings if a prior is not available and a Uniform prior over the same range will be used instead.
Owner
- Name: Colm Talbot
- Login: ColmTalbot
- Kind: user
- Website: https://colmtalbot.github.io/
- Repositories: 6
- Profile: https://github.com/ColmTalbot
GitHub Events
Total
- Release event: 3
- Delete event: 2
- Push event: 2
- Create event: 3
Last Year
- Release event: 3
- Delete event: 2
- Push event: 2
- Create event: 3
Dependencies
- actions/checkout v4 composite
- actions/setup-python v4 composite
- arviz *
- bilby *
- jax *
- numpy *
- numpyro *