edfraenkel / kent_distribution
☆23Updated 4 years ago
Alternatives and similar repositories for kent_distribution
Users that are interested in kent_distribution are comparing it to the libraries listed below
Sorting:
- Implementation of FB8, a generalization of the Kent (1982) and Bingham-Mardia (1978) distributions on a sphere☆18Updated last month
- All things Monte Carlo, written in JAX.☆30Updated 2 years ago
- A differentiable cosmology library in JAX☆196Updated 4 months ago
- Probabilistic Programming and Nested sampling in JAX☆181Updated last month
- A Python toolkit for (simulation-based) inference and the mechanization of science.☆53Updated 3 years ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆93Updated this week
- RKHS feature vectors, operators, and statistical models using JAX for automatic differentiation☆8Updated 3 years ago
- Tools for JAX☆46Updated this week
- Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX☆99Updated last year
- A lightweight and performant implementation of HMC and NUTS in Python, spun out of the PyMC project.☆56Updated 9 months ago
- SBI Workshop jointly by Helmholtz AI + ML ⇌ Science Colaboratory☆23Updated 2 years ago
- A generic interface for linear algebra backends☆73Updated 2 months ago
- A dead simple Python package for creating custom JAX pytree objects☆63Updated 6 months ago
- JAX + Units☆47Updated 3 weeks ago
- Loopy belief propagation for factor graphs on discrete variables, in JAX!☆65Updated 7 months ago
- Minimal Implementation of Bayesian Optimization in JAX☆95Updated 3 weeks ago
- Kernel density estimation on a sphere☆29Updated 3 years ago
- Add a tqdm progress bar to your JAX scans and loops.☆109Updated last week
- Stencil computations in JAX☆71Updated last year
- A lightweight didactic library of kernel methods using the back-end JAX.☆12Updated 2 years ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆103Updated last year
- Documentation:☆119Updated last year
- Probabilistic modeling of tabular data with normalizing flows.☆56Updated 3 weeks ago
- A system for scientific simulation-based inference at scale.☆165Updated last year
- S2FFT: Differentiable and accelerated spherical transforms☆173Updated this week
- 🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and Optax/JAXopt☆61Updated this week
- Object-oriented Jax framework extending Equinox for scientific programming☆16Updated 3 months ago
- A differentiation API for PyTorch☆30Updated 4 years ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆126Updated 8 months ago
- Gaussian Mixture Models in Jax☆18Updated 3 months ago