Tractables / pyjuice
Scalable training and inference for Probabilistic Circuits
☆53Updated last week
Alternatives and similar repositories for pyjuice:
Users that are interested in pyjuice are comparing it to the libraries listed below
- An implementation of EinsumNetworks in PyTorch.☆21Updated last month
- PyTorch implementation for "HyperSPNs: Compact and Expressive Probabilistic Circuits", NeurIPS 2021☆13Updated 3 years ago
- a python framework to build, learn and reason about probabilistic circuits and tensor networks☆88Updated last week
- Loopy belief propagation for factor graphs on discrete variables in JAX☆142Updated 4 months ago
- Lossless compression using Probabilistic Circuits☆16Updated 2 years ago
- Agustinus' very opiniated publication-ready plotting library☆61Updated 2 weeks ago
- This repository holds the code for the NeurIPS 2022 paper, Semantic Probabilistic Layers☆26Updated last year
- Probabilistic Circuits from the Juice library☆104Updated 8 months ago
- ☆19Updated last year
- Code for "Bayesian Structure Learning with Generative Flow Networks"☆83Updated 2 years ago
- A curated collection of papers on probabilistic circuits, computational graphs encoding tractable probability distributions.☆49Updated last year
- ☆50Updated last year
- The Energy Transformer block, in JAX☆56Updated last year
- IVON optimizer for neural networks based on variational learning.☆59Updated 3 months ago
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆31Updated this week
- Graphically structured diffusion model.☆21Updated last year
- A collection of commonly used datasets as benchmarks for density estimation☆24Updated 5 years ago
- Riemannian Optimization Using JAX☆48Updated last year
- Probabilistic programming with HuggingFace language models☆96Updated this week
- Stochastic Automatic Differentiation library for PyTorch.☆197Updated 5 months ago
- ☆12Updated last year
- Sketched matrix decompositions for PyTorch☆69Updated 3 weeks ago
- ☆24Updated this week
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆39Updated 4 years ago
- A Python Library for Deep Probabilistic Modeling☆61Updated 3 months ago
- Squared Non-monotonic Probabilistic Circuits☆21Updated last month
- Pytorch-like dataloaders for JAX.☆75Updated 4 months ago
- PyTorch implementation for "Probabilistic Circuits for Variational Inference in Discrete Graphical Models", NeurIPS 2020☆16Updated 3 years ago
- Code for the paper: "Independent mechanism analysis, a new concept?"☆24Updated last year
- Multiple dispatch over abstract array types in JAX.☆114Updated 2 weeks ago