ximinf

Simulation-Based Inference of Cosmological Parameters in JAX using Type Ia Supernovae.

ximinf is a Python package for performing simulation-based inference (SBI) on cosmological parameters using type Ia supernova (SN Ia) data, built on top of JAX for fast, differentiable, and GPU/TPU-accelerated computation.

Installation

pip install ximinf

Requires Python >= 3.10.

Quickstart

import ximinf

See the documentation for full usage examples covering simulation, training, and inference.

Features

ximinf is organized into five core modules:

ximinf.generate_sim — Generates simulated SN Ia datasets. Includes Latin Hypercube Sampling (LHS) of cosmological/nuisance parameters under configurable priors (uniform, gaussian, half-gaussian, log-uniform, exponential, etc.), and per-dataset simulation via skysurvey.
ximinf.selection_effects — Tools for injecting realistic observational selection effects (e.g. Malmquist bias) into simulated supernova samples via stochastic magnitude-limited detection.
ximinf.nn_train — Training utilities for the neural network classifiers used in the inference step, including loss/accuracy functions, training/validation loops with early stopping, and JAX device setup helpers.
ximinf.nn_test — Diagnostics for validating inference quality, notably TARP (Tests of Accuracy with Random Points) coverage statistics computed across parameter groups.
ximinf.nn_inference — Neural posterior/likelihood-ratio inference. Builds and runs BlackJAX NUTS samplers over grouped parameter blocks, combining trained neural network outputs with analytic log-priors.

Typical workflow

Simulate — Draw parameter sets with generate_sim.scan_params and generate mock SN Ia samples with generate_sim.simulate_one.
(Optional) Apply selection effects — Use selection_effects.apply_malmquist_bias to emulate realistic detection thresholds.
Train — Train per-group neural classifiers on simulated data with nn_train.train_loop.
Infer — Sample the posterior over cosmological parameters with nn_inference.sample_posterior / nn_inference.inference_loop, using a NUTS kernel built on the trained networks.
Validate — Check calibration of the resulting posteriors with nn_test.compute_ecp_tarp_groups.

License

This project is licensed under the GNU General Public License v3.0 (GPLv3). See LICENSE for details.

Author

Adam Trigui — a.trigui@ip2i.in2p3.fr IP2I Lyon

Metadata

Release files for ximinf 0.0.192

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

Source distribution (sdist)

Source distribution for ximinf 0.0.192
File	Size	Uploaded
ximinf-0.0.192.tar.gz	51.4 kB	Oct 14, 2026	Details

Built distribution (wheel)

Table of built distributions (wheels) for ximinf 0.0.192
File	Interpreter	ABI	Platform	Reset
ximinf-0.0.192-py3-none-any.whl 50.2 kB Oct 14, 2026	Python 3	none	any	Details

Total release size: 101.5 kB

Release files / ximinf-0.0.192.tar.gz

Download URL	ximinf-0.0.192.tar.gz
Size	51.4 kB
Tags	Source
SHA-256 checksum How to use checksums	`c4c5aa155030a0fa0df5fcfb1c3a33bb9158f508761632d69ea6860b604b4a6c`
BLAKE2b-256 checksum How to use checksums	`63a324bfc1b7476e81f280d37737d54b425f2c45f2392a7e223ba87dbbec8488`
Upload date	Oct 14, 2026
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/7.0.0 CPython/3.11.17`

Release files / ximinf-0.0.192-py3-none-any.whl

Download URL	ximinf-0.0.192-py3-none-any.whl
Size	50.2 kB
Tags	Python 3
SHA-256 checksum How to use checksums	`c6af0c76eca81283d3f9f0a4ca9c7d13e99c1e2599484f6b1a4f90b0d20eb0fb`
BLAKE2b-256 checksum How to use checksums	`d16b4399696eb9ab71c69b7ec92f16c6968e1b7f6001cffcc9706f4963c82c73`
Upload date	Oct 14, 2026
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/7.0.0 CPython/3.11.17`