A miscellaneous set of helper functions, custom distributions, and other utilities that I find useful when using NumPyro in my work

Differentiable Cosmological Forward Model

Solve automatic numerical differentiation problems in one or more variables.

Neural Network-Boosted Importance Nested Sampling for Bayesian Statistics

OneCovariance code for KiDS to play

A python package to build AI-powered real-time audio applications

Methodology to handle NMF when the data can be negative.

automatic differentiation made easier for C++

A Python wrapper for the lightning fast fast3tree library, a BSP tree implementation written by Peter Behroozi.

FIRst Extragalactic Simulation Of Neutrino and Gamma-ray

Pipeline to compute Nx2pt angular power spectra and their covariances.

Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation

Differentiable and accelerated spherical transforms with JAX

Direct spherical harmonic transform code for point sets on the sphere.

Computer vision assisted tool to extract numerical data from plot images.

Save All Correlations and Covariances

Auto-differentiable methods for Cosmology

differentiable boltzmann code

Python tools for working with the IceCube public data.

A unified pseudo-Cl framework

A Jax implementation to remap cubic volumes with periodic boundary conditions to non-cuboid volumes

Helpful LaTeX files. This repository formerly named "yahapj".

A pure Python halo-model implementation for power spectra of any large-scale structure tracer combination.

A fast, flexible, automated, and differentiable astronomical image 2D forward modelling tool for precise parallel multi-wavelength photometry

A very simple gas regulator model.

Transparent calculations with uncertainties on the quantities involved (aka "error propagation"); calculation of derivatives.

Measurements and emulation of Lagrangian bias models for clustering and lensing cross-correlations.