A toolkit for managing HDF5 simulation data, computing key cosmic quantities, and visualizing results.

JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library

JAX-powered Cosmological Particle-Mesh N-body Solver

Hot Reloading and Profiling for Python

Learning in infinite dimension with neural operators.

A series of tutorial notebooks on denoising diffusion probabilistic models in PyTorch

Sliced Iterative Generator (SIG) & Gaussianizing Iterative Slicing (GIS)

Code for leveraging Information Maximising Neural Networks for optimal cosmological field compression and Bayesian inference for cosmological parameters

Code for plotting bounds on the abundance of primordial black holes (PBHs)

A system for scientific simulation-based inference at scale.

The Julia Programming Language

The Non-Poissonian Template Fit in python/cython: http://nptfit.readthedocs.io/

Proof-of-principle application of Gaussian process modeling to gamma-ray analyses. Code repository associated with the paper https://arxiv.org/abs/2010.10450.