ignaciomagana

pykan

Kolmogorov Arnold Networks

PopModels.jl

Population modeling in Julia.

fmmax

Fourier modal method with Jax

s2fft

Differentiable and accelerated spherical transforms with JAX

gpar

Implementation of the Gaussian Process Autoregressive Regression Model

spectral_sirens

Spectral siren cosmology with the population of compact binary coalescences

interpax

Interpolation and function approximation with JAX

hanabi

Identify and characterize strongly-lensed gravitational waves

COSMIC

COSMIC (Compact Object Synthesis and Monte Carlo Investigation Code)

backpop_sandbox

suave

The Continuous-Function Estimator for 2-point statistics (trash the bins!)

denmarf

Density EstimatioN using Masked AutoRegressive Flow

JAX-GalSim

JAX port of GalSim, for parallelized, GPU accelerated, and differentiable galaxy image simulations.

simsurvey

Simulation for survey (cadence, strategy) based on Python

TGP.pytorch

Repository for the work Transforming Gaussian Processes with Normalizing Flows published at AISTATS 2021

jim

Gravitational-wave data analysis tools in Jax

ridge_map

Ridge plots of ridges

condacolab

Install Conda and friends on Google Colab, easily

diffrax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

GPJax

Gaussian processes in JAX.

lenstronomy-tutorials

Extension modules to the lenstronomy software package

LitePIG

circom

zkSnark circuit compiler

flamethrower

A blazingly fast router for static sites

treeo

A small library for creating and manipulating custom JAX Pytree classes

awesome-zero-knowledge-proofs

A curated list of awesome things related to learning Zero-Knowledge Proofs (ZKP).

emcee-jax

An experiment: emcee implemented in JAX

snarkjs

zkSNARK implementation in JavaScript & WASM

snapjax

a fast, flexible SuperNova Analysis Pipeline for Cosmology in Jax

asdf

ASDF (Advanced Scientific Data Format) is a next generation interchange format for scientific data