Normalizing Flows using JAX
☆87Nov 30, 2023Updated 2 years ago
Alternatives and similar repositories for NuX
Users that are interested in NuX are comparing it to the libraries listed below
Sorting:
- ☆624Jan 23, 2026Updated last month
- Layered distributions using FLAX/JAX☆10Dec 13, 2020Updated 5 years ago
- Documentation:☆129May 22, 2023Updated 2 years ago
- Manifold-learning flows (ℳ-flows)☆233Nov 13, 2020Updated 5 years ago
- Code for the paper "Learning Differential Equations that are Easy to Solve"☆287Nov 25, 2021Updated 4 years ago
- Exponential families for JAX☆76Updated this week
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆105Oct 23, 2023Updated 2 years ago
- Using neural networks to extract sufficient statistics from data by maximising the Fisher information☆32May 12, 2023Updated 2 years ago
- Simple, extensible implementations of some meta-learning algorithms in Jax☆11Oct 6, 2020Updated 5 years ago
- Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.☆330Mar 20, 2024Updated last year
- Fine-grained, dynamic control of neural network topology in JAX.☆21Jul 23, 2023Updated 2 years ago
- Turning SymPy expressions into JAX functions☆45Mar 17, 2021Updated 4 years ago
- BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.☆1,025Feb 3, 2026Updated last month
- Code for paper "SurVAE Flows: Surjections to Bridge the Gap between VAEs and Flows"☆290Feb 1, 2021Updated 5 years ago
- An experiment: emcee implemented in JAX☆28Jul 1, 2022Updated 3 years ago
- Normalizing Flows in Jax☆107Aug 19, 2020Updated 5 years ago
- Probabilistic modeling of tabular data with normalizing flows.☆58Feb 11, 2026Updated 3 weeks ago
- Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.☆241Dec 22, 2023Updated 2 years ago
- Neural likelihood-free methods in PyTorch.☆39Feb 11, 2020Updated 6 years ago
- tutorials on shape measurement☆10Jan 5, 2022Updated 4 years ago
- a differentiable lensing simulator☆10Jan 18, 2022Updated 4 years ago
- Benchmarking field-level cosmological inference from galaxy surveys.☆12Jul 17, 2025Updated 7 months ago
- Implementation of Vision Transformers in Flax☆18Oct 12, 2020Updated 5 years ago
- Simulation-based inference benchmark☆108Jan 23, 2025Updated last year
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆517Updated this week
- Probabilistic Programming and Nested sampling in JAX☆223Feb 27, 2026Updated last week
- A package for creating a matrix product state (MPS) from a free fermion (Gaussian) state.☆10Dec 6, 2021Updated 4 years ago
- E(n) Equivariant GNN in jax☆14Aug 31, 2023Updated 2 years ago
- Normalizing flows in PyTorch☆993Dec 27, 2024Updated last year
- A Python library to integrate automatic differentiation tools such as JAX with QuTiP and related quantum software packages.☆44May 8, 2022Updated 3 years ago
- Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX☆103Jul 6, 2023Updated 2 years ago
- SBI Workshop jointly by Helmholtz AI + ML ⇌ Science Colaboratory☆23Apr 12, 2023Updated 2 years ago
- Code for the research paper "HINT: Hierarchical Invertible Neural Transport for Density Estimation and Bayesian Inference".☆22Feb 11, 2021Updated 5 years ago
- jax library for E3 Equivariant Neural Networks☆224Aug 25, 2025Updated 6 months ago
- ☆80Sep 4, 2021Updated 4 years ago
- Multiple dispatch over abstract array types in JAX.☆137Dec 15, 2025Updated 2 months ago
- Why multiple dispatch lets you write composable code☆40Sep 25, 2020Updated 5 years ago
- A Pytree Module system for Deep Learning in JAX☆213Feb 26, 2023Updated 3 years ago
- ☆21Nov 29, 2022Updated 3 years ago