Tractables / pyjuice
Scalable training and inference for Probabilistic Circuits
☆49Updated last week
Related projects ⓘ
Alternatives and complementary repositories for pyjuice
- ☆18Updated last year
- ☆50Updated last year
- PyTorch implementation for "HyperSPNs: Compact and Expressive Probabilistic Circuits", NeurIPS 2021☆13Updated 3 years ago
- An implementation of EinsumNetworks in PyTorch.☆20Updated last week
- a python framework to build, learn and reason about probabilistic circuits and tensor networks☆80Updated this week
- Probabilistic Circuits from the Juice library☆104Updated 5 months ago
- A curated collection of papers on probabilistic circuits, computational graphs encoding tractable probability distributions.☆48Updated 9 months ago
- Loopy belief propagation for factor graphs on discrete variables in JAX☆131Updated last month
- Graphically structured diffusion model.☆18Updated last year
- A Python Library for Deep Probabilistic Modeling☆60Updated 3 weeks ago
- A collection of commonly used datasets as benchmarks for density estimation☆24Updated 4 years ago
- Agustinus' very opiniated publication-ready plotting library☆58Updated 2 months ago
- An implementation of squared neural families in PyTorch☆11Updated 3 weeks ago
- The Energy Transformer block, in JAX☆53Updated 11 months ago
- Squared Non-monotonic Probabilistic Circuits☆19Updated 4 months ago
- Inference Combinators in JAX☆44Updated 3 weeks ago
- Riemannian Optimization Using JAX☆45Updated last year
- ☆24Updated last year
- This repository holds the code for the NeurIPS 2022 paper, Semantic Probabilistic Layers☆26Updated 11 months ago
- Loopy belief propagation for factor graphs on discrete variables, in JAX!☆64Updated last month
- Lossless compression using Probabilistic Circuits☆16Updated 2 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆37Updated 4 years ago
- Stochastic Automatic Differentiation library for PyTorch.☆180Updated 2 months ago
- Code for UAI'19: Random Sum-Product Networks: A Simple and Effective Approach to Probabilistic Deep Learning☆36Updated 4 years ago
- A generic interface for linear algebra backends☆70Updated 4 months ago
- Multiple dispatch over abstract array types in JAX.☆105Updated last week
- Sketched matrix decompositions for PyTorch☆67Updated 6 months ago
- ☆54Updated 2 years ago
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆25Updated 2 months ago
- IVON optimizer for neural networks based on variational learning.☆53Updated 2 weeks ago