Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX
☆216Aug 3, 2025Updated 7 months ago
Alternatives and similar repositories for torchquad
Users that are interested in torchquad are comparing it to the libraries listed below
Sorting:
- Abstract your array operations.☆164Updated this week
- Source code for paper "Learning the Solution Operator of Boundary Value Problems using Graph Neural Networks"☆19Jul 25, 2024Updated last year
- Gradient Based Nested Sampling☆21Feb 2, 2024Updated 2 years ago
- ☆11Oct 16, 2023Updated 2 years ago
- Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/☆1,919Feb 23, 2026Updated last week
- Gradient-informed particle MCMC methods☆12Jan 29, 2024Updated 2 years ago
- The tiniest of Gaussian Process libraries☆333Updated this week
- Python interface to the QDLDL (https://github.com/osqp/qdldl) free LDL factorization routine for quasi-definite linear systems☆16Feb 19, 2026Updated 2 weeks ago
- BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.☆1,021Feb 3, 2026Updated last month
- A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods☆1,553May 2, 2024Updated last year
- A zoo of implementations of differential equation problems in NumPy and JAX. Oscillators, chemical reactions, n-body problems, epidemiolo…☆17Dec 28, 2023Updated 2 years ago
- Python package that numerically calculates the structure of probability distributions on statistical manifolds.☆15Feb 24, 2025Updated last year
- Free-form flows are a generative model training a pair of neural networks via maximum likelihood☆50Jun 26, 2025Updated 8 months ago
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆26Dec 21, 2023Updated 2 years ago
- A JAX-based gaussian process structure learning package for time series modelling.☆36May 28, 2025Updated 9 months ago
- The MaNGA Data Analysis Pipeline☆21Jul 2, 2025Updated 8 months ago
- Fast, lightweight and parallelised simulation-based inference in JAX.☆23Oct 28, 2025Updated 4 months ago
- An omni-sampler for Bayesian MCMC☆58Feb 25, 2026Updated last week
- Algorithms for computations on random manifolds made easier☆94Dec 4, 2023Updated 2 years ago
- Normalizing flows in PyTorch☆444Dec 1, 2025Updated 3 months ago
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆67Mar 11, 2024Updated last year
- Subsumed into xnd☆25Aug 30, 2023Updated 2 years ago
- A Structural Average of Labeled Merge Trees for Uncertainty Visualization☆11Apr 22, 2024Updated last year
- A Latex Template for a Science Textbook☆10Apr 7, 2020Updated 5 years ago
- Source code repository for the AISTAT 2023 paper Transport Reversible Jump Proposals.☆10Mar 3, 2023Updated 3 years ago
- ☆219Feb 21, 2026Updated last week
- VegasFlow: accelerating Monte Carlo simulation across multiple hardware platforms☆37Apr 15, 2025Updated 10 months ago
- Likelihood-free AMortized Posterior Estimation with PyTorch☆133Aug 21, 2024Updated last year
- Numerical quadrature with JAX☆77Feb 16, 2026Updated 2 weeks ago
- Conjugate Gradient related code for "Solver-in-the-Loop: Learning from Differentiable Physics to Interact with Iterative PDE-Solvers"☆24Dec 9, 2020Updated 5 years ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆546Feb 16, 2026Updated 2 weeks ago
- Reproduction of the paper SFSRNet: Super-resolution for single-channel Audio Source Separation by me (@arda-num) and @dritx16. Navigate P…☆11Jul 7, 2022Updated 3 years ago
- The Online MCMC☆10Feb 14, 2025Updated last year
- ☆11Oct 15, 2023Updated 2 years ago
- ☆11Sep 2, 2024Updated last year
- Numerical relativity surrogate waveform in Jax☆19Aug 14, 2025Updated 6 months ago
- Individual binary analysis of the NANOGrav 15-year dataset☆10Jun 26, 2023Updated 2 years ago
- Richard @rdgao & Michael @michaeldeistler: using neural network-based regression and density estimation for Generalized Bayesian Inferenc…☆11Dec 19, 2025Updated 2 months ago
- Codes to reproduce results in the paper☆10Apr 9, 2021Updated 4 years ago