General analysis tools for the DC2 Data Set.

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Add a tqdm progress bar to your JAX scans and loops.

The tiniest of Gaussian Process libraries

JAX - A curated list of resources https://github.com/google/jax

Differentiable Likelihood for CMB Analysis

BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.

Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python ⚡

A differentiable cosmology library in JAX

A common reader interface for accessing generic catalogs

A Jax-based library for designing and training transformer models from scratch.

Flax is a neural network library for JAX that is designed for flexibility.

SPT-SZ cluster cosmology likelihood code and data to reproduce Bocquet et al. 2019

MLX: An array framework for Apple silicon

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Automated generation of NASA ADS bibtex entries directly from citation keys in your TeX source files

Repository containing the material for running the Capse.jl emulator and reproduce all the results of the paper.

Examples for usage of the Core Cosmology Library

Optax is a gradient processing and optimization library for JAX.

Code to calculate the CIB, tSZ, and CIBxtSZ power spectra using a halo model

A rectangular pixel map manipulation and harmonic analysis library derived from Sigurd Naess' enlib.

Cluster cosmology analysis using DESC tools and simulations

DESC Core Cosmology Library: cosmology routines with validated numerical accuracy

SO Likelihoods and Theories

Machine Learning - accelerated Bayesian inference

Emulator for the halo mass function (HMF) from the Mira-Titan Universe suite of cosmological N-body simulations.

R-package for Bayesian linear regression in astronomy