thaler-lab / Wasserstein
Python/C++ library for computing Wasserstein distances efficiently.
☆28Updated 4 months ago
Alternatives and similar repositories for Wasserstein
Users that are interested in Wasserstein are comparing it to the libraries listed below
Sorting:
- S2FFT: Differentiable and accelerated spherical transforms☆150Updated this week
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆67Updated 2 weeks ago
- A small library of paramaterizations and parameter constraints for PyTrees.☆29Updated last week
- Probabilistic Programming and Nested sampling in JAX☆179Updated last month
- JAX + Units☆47Updated 2 weeks ago
- Provides differentiable versions of common HEP operations and objectives.☆24Updated last year
- Stencil computations in JAX☆70Updated last year
- Vector classes and utilities☆88Updated this week
- Universal Histogram Interface☆18Updated last week
- A differentiable cosmology library in JAX☆196Updated 4 months ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆126Updated 7 months ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆92Updated this week
- differentiable (binned) likelihoods with JAX☆22Updated this week
- Proximal Nested Sampling for high-dimensional Bayesian model selection☆23Updated last year
- Automatic parametric modeling with symbolic regression☆46Updated last week
- Differentiable and gpu enabled fast wavelet transforms in JAX.☆42Updated 10 months ago
- JAX bindings for the NVIDIA cuDecomp library☆35Updated last week
- Scientific numbers with multiple uncertainties, correlation-aware gaussian error propagation and numpy support.☆28Updated 4 months ago
- Manipulating ragged arrays in an Array API compliant way.☆41Updated 2 weeks ago
- Princeton mini course on GPUs in Python☆40Updated 6 months ago
- Upstream optimisation for downstream inference☆69Updated last month
- Python implementation of the BumpHunter algorithm used by HEP community.☆17Updated 8 months ago
- Statistics tools and utilities.☆75Updated 2 weeks ago
- Minimal Implementation of Bayesian Optimization in JAX☆95Updated 2 weeks ago
- Numerical derivatives for Python☆44Updated last week
- All things Monte Carlo, written in JAX.☆30Updated 2 years ago
- Machine learning–based inference toolkit for particle physics☆86Updated 6 months ago
- Python/numba package for evaluating Wigner D matrices and spin-weighted spherical harmonics☆54Updated 2 weeks ago
- IterGP: Computation-Aware Gaussian Process Inference (NeurIPS 2022)☆41Updated 2 years ago