Skip to main content

aspire: Accelerated Sequential Posterior Inference via REuse

DOI PyPI Documentation Status tests

aspire is a framework for reusing existing posterior samples to obtain new results at a reduced cost.

Installation

aspire can be installed from PyPI using pip. By default, you need to install one of the backends for the normalizing flows, either torch or jax. We also recommend installing minipcn if using the smc sampler:

Torch

We recommend installing torch manually to ensure correct CPU/CUDA versions are installed. See the PyTorch installation instructions for more details.

pip install aspire-inference[torch,minipcn]

Jax:

We recommend install jax manually to ensure the correct GPU/CUDA versions are installed. See the jax documentation for details

pip install aspire-inference[jax,minipcn]

Important: the name of aspire on PyPI is aspire-inference but once installed the package can be imported and used as aspire.

Quickstart

import numpy as np
from aspire import Aspire, Samples

# Define a log-likelihood and log-prior
def log_likelihood(samples):
    x = samples.x
    return -0.5 * np.sum(x**2, axis=-1)

def log_prior(samples):
    return -0.5 * np.sum(samples.x**2, axis=-1)

# Create the initial samples
init = Samples(np.random.normal(size=(2_000, 4)))

# Define the aspire object
aspire = Aspire(
    log_likelihood=log_likelihood,
    log_prior=log_prior,
    dims=4,
    parameters=[f"x{i}" for i in range(4)],
)

# Fit the normalizing flow
aspire.fit(init, n_epochs=20)

# Sample the posterior
posterior = aspire.sample_posterior(
    sampler="smc",
    n_samples=500,
    sampler_kwargs=dict(n_steps=100),
)

# Plot the posterior distribution
posterior.plot_corner()

Documentation

See the documentation on ReadTheDocs.

Citation

If you use aspire in your work please cite the DOI and paper.

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

aspire_inference-0.1.0.tar.gz (91.9 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

aspire_inference-0.1.0-py3-none-any.whl (67.8 kB view details)

Uploaded Python 3

File details

Details for the file aspire_inference-0.1.0.tar.gz.

File metadata

  • Download URL: aspire_inference-0.1.0.tar.gz
  • Upload date:
  • Size: 91.9 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/7.0.0 CPython/3.13.14

File hashes

Hashes for aspire_inference-0.1.0.tar.gz
Algorithm Hash digest
SHA256 679c0ddd8cd6f4918b55b8f1b8e38b3a8be2baa659cf46cf61578eb6e5adff83
MD5 184b38db97792b11c94dfca20acfad59
BLAKE2b-256 9fe166ac439f3f7a06fe23c9fa0eab6247cf4f48ff3d2956e5d967d2508397b1

See more details on using hashes here.

Provenance

The following attestation bundles were made for aspire_inference-0.1.0.tar.gz:

Publisher: publish.yml on mj-will/aspire

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file aspire_inference-0.1.0-py3-none-any.whl.

File metadata

File hashes

Hashes for aspire_inference-0.1.0-py3-none-any.whl
Algorithm Hash digest
SHA256 c7fd738d423b5eef8000de0f8fe41ff459ff0dd926333aeb1182adbe1eb2ea11
MD5 c3afb4f0c48c1cda3016f2c6e3b982a4
BLAKE2b-256 26befc2924aa70b08fc7ce2555c290cddea18bf9b03695a681a54453cd66f5b4

See more details on using hashes here.

Provenance

The following attestation bundles were made for aspire_inference-0.1.0-py3-none-any.whl:

Publisher: publish.yml on mj-will/aspire

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page