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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆658Oct 20, 2026Updated 2 weeks ago
- Generative models with JAX.☆22Feb 21, 2026Updated 8 months ago
- Documentation:☆132May 22, 2023Updated 3 years ago
- Manifold-learning flows (ℳ-flows)☆238Nov 13, 2020Updated 5 years ago
- Code for the paper "Learning Differential Equations that are Easy to Solve"☆291Nov 25, 2021Updated 4 years ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Code for paper "SurVAE Flows: Surjections to Bridge the Gap between VAEs and Flows"☆292Feb 1, 2021Updated 5 years ago
- ☆21Nov 29, 2022Updated 3 years ago
- Probabilistic modeling of tabular data with normalizing flows.☆64May 6, 2026Updated 6 months ago
- Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.☆331Mar 20, 2024Updated 2 years ago
- Benchmarking field-level cosmological inference from galaxy surveys.☆16Jul 17, 2025Updated last year
- Neural likelihood-free methods in PyTorch.☆39Feb 11, 2020Updated 6 years ago
- We got a stew going!☆27Oct 3, 2023Updated 3 years ago
- Normalizing Flows in Jax☆109Aug 19, 2020Updated 6 years ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆106Oct 23, 2023Updated 3 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Implementation of Vision Transformers in Flax☆18Oct 12, 2020Updated 6 years ago
- A package for creating a matrix product state (MPS) from a free fermion (Gaussian) state.☆10Dec 6, 2021Updated 4 years ago
- BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.☆1,138Updated this week
- Multiple dispatch over abstract array types in JAX.☆143Updated this week
- Exponential families for JAX☆77Updated this week
- Turning SymPy expressions into JAX functions☆45Mar 17, 2021Updated 5 years ago
- A differentiable cosmology library in JAX☆248Jun 27, 2025Updated last year
- Using neural networks to extract sufficient statistics from data by maximising the Fisher information☆34May 12, 2023Updated 3 years ago
- Graph Convolutional Networks in JAX☆34Jan 3, 2021Updated 5 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Normalizing flows in PyTorch☆1,020Dec 27, 2024Updated last year
- The official code for the Harmonic variational Gaussian processes (ICML2021).☆11Jun 6, 2021Updated 5 years ago
- An experiment: emcee implemented in JAX☆28Jul 1, 2022Updated 4 years ago
- A Python library to integrate automatic differentiation tools such as JAX with QuTiP and related quantum software packages.☆44May 8, 2022Updated 4 years ago
- Simple, extensible implementations of some meta-learning algorithms in Jax☆11Oct 6, 2020Updated 6 years ago
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆550Oct 26, 2026Updated last week
- SBI Workshop jointly by Helmholtz AI + ML ⇌ Science Colaboratory☆25Apr 12, 2023Updated 3 years ago
- ☆80Sep 4, 2021Updated 5 years ago
- a differentiable lensing simulator☆12Jan 18, 2022Updated 4 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- JAXChem is a JAX-based deep learning library for complex and versatile chemical modeling☆85Jul 15, 2020Updated 6 years ago
- Meeting repo for likelihood free inference meeting☆14Oct 6, 2022Updated 4 years ago
- Meta-VQE data and examples repository☆11Oct 13, 2020Updated 6 years ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆48Aug 18, 2026Updated 2 months ago
- A Julia code for performing variational Monte Carlo (VMC) simulations of determinantal wave functions☆15Nov 16, 2017Updated 8 years ago
- Layered distributions using FLAX/JAX☆10Dec 13, 2020Updated 5 years ago
- code for "Residual Flows for Invertible Generative Modeling".☆276Sep 19, 2023Updated 3 years ago