Metadata-Version: 2.4
Name: cosmulator
Version: 0.0.35
Summary: Train, validate and serve machine-learning emulators of posterior distributions
Author-email: Dily Duan Yi Ong <dlo26@cam.ac.uk>
License-Expression: MIT
Project-URL: Homepage, https://github.com/DilyOng/cosmulator
Project-URL: Bug Tracker, https://github.com/DilyOng/cosmulator/issues
Project-URL: Documentation, https://cosmulator.readthedocs.io
Classifier: Development Status :: 3 - Alpha
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: 3.13
Classifier: Operating System :: OS Independent
Classifier: Intended Audience :: Science/Research
Classifier: Topic :: Scientific/Engineering
Classifier: Topic :: Scientific/Engineering :: Astronomy
Classifier: Topic :: Scientific/Engineering :: Physics
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.11
Description-Content-Type: text/x-rst
License-File: LICENSE
Requires-Dist: numpy
Requires-Dist: scipy
Requires-Dist: requests
Provides-Extra: train
Requires-Dist: jax; extra == "train"
Requires-Dist: optax; extra == "train"
Requires-Dist: flowjax; extra == "train"
Requires-Dist: equinox; extra == "train"
Provides-Extra: hpo
Requires-Dist: optuna; extra == "hpo"
Provides-Extra: docs
Requires-Dist: sphinx; extra == "docs"
Requires-Dist: sphinx_rtd_theme; extra == "docs"
Requires-Dist: numpydoc; extra == "docs"
Provides-Extra: test
Requires-Dist: pytest; extra == "test"
Requires-Dist: pytest-cov; extra == "test"
Requires-Dist: flake8; extra == "test"
Requires-Dist: pydocstyle; extra == "test"
Requires-Dist: packaging; extra == "test"
Requires-Dist: pre-commit; extra == "test"
Dynamic: license-file

=============================================================================================
cosmulator: accelerating cosmological inference with JAX machine-learning emulators on GPU
=============================================================================================

:Author: Dily Duan Yi Ong
:License: MIT
:Homepage: https://github.com/DilyOng/cosmulator

``cosmulator`` trains JAX normalising-flow emulators of nested-sampling
posteriors on GPU, so a distribution can be resampled instantly for fast
marginalisation and Bayesian evidence, and manages the resulting emulators
(training, validation, storage and serving) across a grid of cosmological
models and surveys.

Developed under a UKRI-funded high performance computing (DiRAC) project.

Installation
------------

.. code:: bash

    pip install cosmulator            # inspect and use emulators (numpy only)
    pip install "cosmulator[train]"   # adds the JAX training stack

Requires Python 3.11 or later.
