Exact OU processes with JAX
☆59Jan 6, 2026Updated 2 months ago
Alternatives and similar repositories for thermox
Users that are interested in thermox are comparing it to the libraries listed below
Sorting:
- Uncertainty quantification with PyTorch☆378Mar 10, 2026Updated last week
- ☆13Dec 20, 2024Updated last year
- A full-stack, cross-platform software for generating, compiling, and executing quantum circuits for simulating quantum many-body dynamics…☆17Jun 2, 2021Updated 4 years ago
- Code to reproduce experiments in Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flows☆13May 23, 2024Updated last year
- A C++ framework designed for computer graphics software☆12Jan 31, 2026Updated last month
- In which I learn about score functions and how they can be used to generate data.☆16Apr 5, 2024Updated last year
- Jaxpr Visualisation Tool☆36Dec 22, 2024Updated last year
- Berkeley Single Cell Computational Microscopy dataset☆18Oct 27, 2025Updated 4 months ago
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆317Updated this week
- Code for "Invariance Learning in Deep Neural Networks with Differentiable Laplace Approximations"☆23Nov 16, 2022Updated 3 years ago
- ☆10Mar 2, 2022Updated 4 years ago
- Code for the talk "Approximate Bayesian inference for latent Gaussian models in Stan"☆13May 18, 2022Updated 3 years ago
- Code for NeurIPS 2024 paper: "Noether's razor: Learning Conserved Quantities" by Tycho F. A. van der Ouderaa, Mark van der Wilk, Pim de H…☆10Oct 12, 2024Updated last year
- Sum and cumulative sum using the Kahan-Babuska-Neumaier algorithm☆24Mar 9, 2026Updated last week
- ☆18Nov 10, 2023Updated 2 years ago
- Testing the performance of different tensor network architectures on MNIST classification tasks.☆11Jun 1, 2020Updated 5 years ago
- ☆15Updated this week
- ☆21Jan 19, 2024Updated 2 years ago
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.☆14Dec 21, 2021Updated 4 years ago
- Add a tqdm progress bar to your JAX scans and loops.☆125May 9, 2025Updated 10 months ago
- The Energy Transformer block, in JAX☆64Dec 14, 2023Updated 2 years ago
- hybrid classical-quantum transfer learning for cardiomegaly detection on chest x-rays☆13Jul 12, 2023Updated 2 years ago
- TNCO is a heuristic tool that optimizes tensor network contraction paths.☆32Mar 14, 2026Updated last week
- C++ Library for Quantum State Preparation (QSP)☆12Jan 5, 2023Updated 3 years ago
- Material for my proposed StanCon talk☆20Oct 15, 2020Updated 5 years ago
- Stainless neural networks in JAX☆34Feb 3, 2026Updated last month
- Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/☆1,933Feb 23, 2026Updated 3 weeks ago
- Koans to learn rhine☆13Mar 10, 2026Updated last week
- ☆14Oct 7, 2022Updated 3 years ago
- Bayesian model reduction for probabilistic machine learning☆11Jul 3, 2025Updated 8 months ago
- Using JAX to generate piano music as MIDI☆39Nov 28, 2023Updated 2 years ago
- Mathematical operations for JAX pytrees☆209Dec 5, 2024Updated last year
- Write great documents with markdown, then execute in the shell.☆10Sep 1, 2017Updated 8 years ago
- A metrics library for the JAX ecosystem☆41Mar 16, 2023Updated 3 years ago
- A comprehensive JAX/NNX library for diffusion and flow matching generative algorithms, featuring DiT (Diffusion Transformer) and its vari…☆139Oct 16, 2025Updated 5 months ago
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆304Updated this week
- Matrix-free linear algebra in JAX.☆161Dec 2, 2025Updated 3 months ago
- BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.☆1,033Feb 3, 2026Updated last month
- Simulation-based inference in JAX☆46Mar 13, 2026Updated last week