cosmoforge 1.0.0rc2

CosmoForge: umbrella metapackage installing cosmocore, qube-qml, and picslike.

CosmoForge

📚 Complete Documentation | Installation Guide | Quick Start | API Reference

CosmoForge is a comprehensive Python framework for power spectrum estimation and likelihood analysis of spin-0 and spin-2 fields on the sphere, using Fisher matrix, Quadratic Maximum Likelihood (QML), and pixel-based likelihood methods. While widely applicable to any sky signal (e.g. CMB, galaxy surveys, 21 cm), it is particularly optimized for the analysis of partial-sky, noisy observations with complex noise covariance.

Overview

CosmoForge consists of several interconnected packages designed for efficient and accurate cosmological parameter estimation:

• cosmocore: Core functionality for cosmological analysis including field management, matrix operations, and I/O utilities
• qube-qml: QML and Fisher matrix implementations for power spectrum estimation (imported as qube)
• picslike: Pixel-based Inference with Correlated-Skies Likelihood — pixel-space likelihood analysis for parameter estimation

Features

• Fisher Matrix Analysis: Fast Fisher matrix computation for cosmological parameter forecasting
• QML Power Spectrum Estimation: Quadratic Maximum Likelihood estimation for optimal power spectrum recovery
• Pixel-Based Likelihood: Direct likelihood evaluation in map pixel space for parameter estimation
• MPI Parallelization: Efficient parallel computation support for large-scale analyses
• HEALPix Integration: Full support for HEALPix pixelization scheme
• Flexible Field Management: Support for scalar (spin-0) and tensor (spin-2) fields
• Beam and Noise Modeling: Comprehensive instrumental effects modeling

Installation

📖 For detailed installation instructions, see the Installation Guide

Requirements

• Python 3.11–3.13
• NumPy, SciPy
• healpy
• mpi4py (for parallel computation)

Install from PyPI

The umbrella distribution installs all three subpackages:

pip install cosmoforge

Or pick subpackages individually:

pip install cosmocore       # core utilities only
pip install qube-qml        # adds QML / Fisher (imported as `qube`)
pip install picslike        # adds pixel-space likelihood

Install from source

git clone https://github.com/ggalloni/CosmoForge.git
cd CosmoForge
uv sync --all-packages --all-extras --dev

Quick Start

🚀 For comprehensive tutorials and examples, visit the Quick Start Guide and Tutorials

Fisher Matrix Analysis

from qube import Fisher

# Initialize Fisher analysis
fisher = Fisher("config/fisher_config.yaml")
fisher.run()

# Get Fisher matrix
fisher_matrix = fisher.get_fisher_matrix()

QML Power Spectrum Estimation

from qube import Spectra

# Initialize QML analysis
qml = Spectra("config/qml_config.yaml")
qml.run()

# Get power spectra
power_spectra = qml.get_power_spectra()
noise_bias = qml.get_noise_bias()

Pixel-Based Likelihood Analysis

from picslike import PICSLike

# Initialize pixel-based likelihood
picslike = PICSLike(params_file="config/picslike_config.yaml")
picslike.run()

# Get results
chi_squared = picslike.get_chi_squared()
best_fit = picslike.get_best_fit()

Using with Precomputed Fisher

from qube import Fisher, Spectra

# Compute Fisher matrix first
fisher = Fisher("config/fisher_config.yaml")
fisher.run()

# Reuse Fisher computation for QML
qml = Spectra("config/qml_config.yaml", fisher=fisher)
qml.run()

Package Structure

CosmoForge/
├── src/
│   ├── cosmoforge.cosmocore/    # Core functionality (published as `cosmocore`)
│   ├── cosmoforge.qube/         # QML and Fisher analysis (published as `qube-qml`, imported as `qube`)
│   ├── cosmoforge.picslike/     # Pixel-space likelihood (published as `picslike`)
│   └── cosmoforge.meta/         # Umbrella metapackage (published as `cosmoforge`)
├── tests/                       # Test suite
├── docs/                        # Documentation
└── examples/                    # Example configurations

Configuration

CosmoForge uses YAML configuration files to specify analysis parameters:

# Example configuration
nside: 4
lmax: 16
fields: "TEB"
maskfile: "data/mask.fits"
inputclfile: "data/fiducial_cls.txt"
# ... additional parameters

Testing

Run the test suite:

# Run all tests
uv run pytest

# Run specific package tests
uv run --package cosmocore pytest src/cosmoforge.cosmocore/tests/
uv run --package qube pytest src/cosmoforge.qube/tests/
uv run --package picslike pytest src/cosmoforge.picslike/tests/

Performance

CosmoForge is designed for high-performance cosmological analysis:

• Numba JIT compilation for critical mathematical operations
• MPI parallelization for distributed computing
• Optimized matrix operations using LAPACK/BLAS
• Memory-efficient algorithms for large datasets

Contributing

1. Fork the repository
2. Create a feature branch
3. Make your changes
4. Add tests for new functionality
5. Submit a pull request

License

[Add license information]

Citation

If you use CosmoForge in your research, please cite:

[Add citation information]

Support

📖 Complete documentation is available at: https://ggalloni.github.io/CosmoForge/

For questions and support:

• Open an issue on GitHub
• Contact: [contact information]

Acknowledgments

CosmoForge builds upon established cosmological analysis methods and libraries:

• HEALPix for pixelization
• NumPy/SciPy for numerical computations
• MPI for parallelization