Skip to main content

astroemu — Next Generation Emulators for Cosmology and Astrophysics

version docs license

Under active development — interfaces may change between versions.

astroemu is a generalised framework for emulating spectral signals in cosmology and astrophysics, inspired by the globalemu package. Neural network emulators are implemented in JAX, with an optax-based training loop and PyTorch-style dataloaders.

Installation

pip install astroemu

How it works

The core idea (shared with globalemu) is to input independent variables alongside physical model parameters, predicting a single spectral value per call. Full spectra are recovered via a vectorised call over all $x$ points.

Given a signal $y = f(x, \theta)$ with $N$ parameter samples and $m$ independent variable points, the training data is tiled so that parameters and independent variables are concatenated as inputs:

Input Output
$[\theta_0,\ x_0]$ $y_0$
$[\theta_0,\ x_1]$ $y_1$
$[\theta_0,\ \ldots]$ $\ldots$
$[\theta_0,\ x_m]$ $y_m$
$[\ldots,\ \ldots]$ $\ldots$
$[\theta_N,\ x_m]$ $y_m$

For more details see the globalemu paper. A paper demonstrating applications of astroemu to a broad range of astrophysical signals is in preparation.

Documentation

git clone git@github.com:htjb/astroemu.git
pip install ".[docs]"
mkdocs serve

or browse the hosted docs at astroemu.readthedocs.io.

Contributions

Contributions are welcome! Please open an issue to discuss and read the contribution guidelines before submitting a pull request.

Metadata

Release files for astroemu 0.4.0

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for astroemu 0.4.0
File Size Uploaded Provenance Actions
astroemu-0.4.0.tar.gz 19.5 kB Without provenance Details for astroemu-0.4.0.tar.gz

Built distribution (wheel)

Table of built distributions (wheels) for astroemu 0.4.0
File Interpreter ABI Platform Actions
astroemu-0.4.0-py3-none-any.whl Python 3 none any Details for astroemu-0.4.0-py3-none-any.whl

Total release size: 34.6 kB

Release files / astroemu-0.4.0.tar.gz

Download URL astroemu-0.4.0.tar.gz
Size 19.5 kB
Tags Source
SHA-256 checksum
How to use checksums
a943b77d147688762fb2cfbba65356d8a3ee157cc58afd85c2f965bb043e2cf6
BLAKE2b-256 checksum
How to use checksums
a5e33518c4c7dc5d7377a6ae74b29f6c94f892c08f7dcf26618d717313dbfd19
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/6.2.0 CPython/3.11.11

Release files / astroemu-0.4.0-py3-none-any.whl

Download URL astroemu-0.4.0-py3-none-any.whl
Size 15.1 kB
Tags Python 3
SHA-256 checksum
How to use checksums
1caf4d28014dce203363c25646d27d71796c3d171315748fd1da9bc245284bbd
BLAKE2b-256 checksum
How to use checksums
627b603d16a5b31f2c7c25e6e80d9b739a94d870babbb296cd14721978404667
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/6.2.0 CPython/3.11.11

Security for astroemu 0.4.0

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

No provenance found

None of the files in this release were uploaded with provenance metadata.

A note on package security

PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.

To learn more about protecting yourself from malicious actors, visit our package security documentation.

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page