pierreglaser / jax-tutorialLinks
☆16Updated 3 years ago
Alternatives and similar repositories for jax-tutorial
Users that are interested in jax-tutorial are comparing it to the libraries listed below
Sorting:
- ☆21Updated 3 months ago
- SBI Workshop jointly by Helmholtz AI + ML ⇌ Science Colaboratory☆23Updated 2 years ago
- Combination of transformers and diffusion models for flexible all-in-one simulation-based inference☆79Updated 7 months ago
- Simulation-based inference in JAX☆44Updated 2 months ago
- Conditional density estimation with neural networks☆35Updated last year
- MCHMC: sampler from an arbitrary differentiable distribution☆75Updated 2 months ago
- Community-sourced list of papers and resources on neural simulation-based inference.☆146Updated last week
- Benchopt benchmark for Lasso☆14Updated 6 months ago
- Richard @rdgao & Michael @michaeldeistler: using neural network-based regression and density estimation for Generalized Bayesian Inferenc…☆11Updated last month
- Simulation-based inference benchmark☆108Updated last year
- A generic library for linear and non-linear Gaussian smoothing problems. The code leverages JAX and implements several linearization algo…☆12Updated last year
- Laplace approximations in JAX.☆43Updated this week
- Probabilistic programming framework for signal inference algorithms that operate regardless of the underlying grids and their resolutions☆41Updated last week
- A system for scientific simulation-based inference at scale.☆165Updated last year
- Differentiable and gpu enabled fast wavelet transforms in JAX.☆48Updated last year
- Density estimation likelihood-free inference. No longer actively developed see https://github.com/mackelab/sbi instead☆73Updated 5 years ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆114Updated this week
- Hierarchical neural implicit inference over event ensembles. Code repository associated with https://arxiv.org/abs/2306.12584.☆13Updated 2 years ago
- ☆11Updated 2 years ago
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆255Updated 3 months ago
- Probabilistic modeling of tabular data with normalizing flows.☆58Updated 3 weeks ago
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆53Updated 6 months ago
- Fast, lightweight and parallelised simulation-based inference in JAX.☆23Updated 3 months ago
- Density Estimation Likelihood-Free Inference with neural density estimators and adaptive acquisition of simulations☆111Updated 2 years ago
- Generative models with JAX.☆20Updated last year
- ☆12Updated 2 years ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆75Updated 4 months ago
- Bayesian learning and inference for state space models (SSMs) using Google Research's JAX as a backend☆63Updated last year
- Differentiable neuron simulations with biophysical detail on CPU, GPU, or TPU.☆177Updated this week
- A JAX-based gaussian process structure learning package for time series modelling.☆34Updated 8 months ago