thaler-lab / Wasserstein
Python/C++ library for computing Wasserstein distances efficiently.
☆22Updated this week
Related projects: ⓘ
- Probabilistic Programming and Nested sampling in JAX☆141Updated last month
- Vector classes and utilities☆77Updated this week
- Provides differentiable versions of common HEP operations and objectives.☆24Updated last year
- A differentiable cosmology library in JAX☆172Updated last month
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆61Updated 2 months ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆77Updated last week
- Differentiable and accelerated spherical transforms with JAX☆123Updated this week
- differentiable (binned) likelihoods with JAX☆18Updated 3 weeks ago
- Using neural networks to extract sufficient statistics from data by maximising the Fisher information☆29Updated last year
- Machine learning–based inference toolkit for particle physics☆85Updated last week
- Variational Autoencoder for Particle Physics using Earth-Movers Distance☆13Updated 2 years ago
- Numerical derivatives for Python☆42Updated this week
- Numerical quadrature with JAX☆39Updated last month
- Tutorial Code for MLHEP pyprob☆18Updated last year
- A spherical convolutional neural network for cosmology (TFv2).☆19Updated 5 months ago
- For developing and reproducing ML + HEP projects.☆22Updated 2 weeks ago
- Probabilistic modeling of tabular data with normalizing flows.☆54Updated 3 months ago
- 🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and JAXopt☆45Updated this week
- A package to describe amortized (conditional) normalizing-flow PDFs defined jointly on tensor products of manifolds with coverage control…☆42Updated 7 months ago
- JAX + Units☆41Updated last week
- Universal Histogram Interface☆15Updated this week
- Probabilistic programming framework for signal inference algorithms that operate regardless of the underlying grids and their resolutions☆16Updated this week
- My CV☆36Updated this week
- Jet tagging in the Lund plane with graph networks☆8Updated 2 years ago
- ☆31Updated 3 years ago
- Template for talks in remark+KaTeX.☆25Updated 4 years ago
- Proximal Nested Sampling for high-dimensional Bayesian model selection☆20Updated 5 months ago
- Things that make me feel productive☆15Updated last year
- Upstream optimisation for downstream inference☆69Updated last month
- Inference of substructure properties in strong lensing systems with machine learning. Code repository associated with https://arxiv.org/a…☆30Updated 4 years ago