mcbal / afem
Implementation of approximate free-energy minimization in PyTorch
☆19Updated 3 years ago
Alternatives and similar repositories for afem:
Users that are interested in afem are comparing it to the libraries listed below
- u-MPS implementation and experimentation code used in the paper Tensor Networks for Probabilistic Sequence Modeling (https://arxiv.org/ab…☆19Updated 4 years ago
- Deterministic particle dynamics for simulating Fokker-Planck probability flows☆24Updated 2 years ago
- A minimal implementation of a VAE with BinConcrete (relaxed Bernoulli) latent distribution in TensorFlow.☆22Updated 5 years ago
- Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX☆39Updated last year
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 3 years ago
- Differentiable Euler Characteristic Transform☆17Updated 10 months ago
- PyTorch implementation for "Probabilistic Circuits for Variational Inference in Discrete Graphical Models", NeurIPS 2020☆17Updated 3 years ago
- Riemannian Optimization Using JAX☆48Updated last year
- Implementation of deep implicit attention in PyTorch☆65Updated 3 years ago
- Implementation of Action Matching for the Schrödinger equation☆24Updated last year
- ☆52Updated 2 years ago
- orbital MCMC☆10Updated 3 years ago
- ☆34Updated 2 years ago
- Hierarchical variational models for physics.☆17Updated 4 years ago
- Pytorch code for Sampling in Combinatorial Spaces with SurVAE Flow Augmented MCMC☆12Updated 4 years ago
- Stochastic Normalizing Flows☆76Updated 3 years ago
- ☆19Updated 3 years ago
- Implementation of Action Matching☆41Updated last year
- Website for the ICML 2021 tutorial on Random Matrix Theory and Machine Learning☆15Updated 3 years ago
- ☆46Updated last week
- Exact OU processes with JAX☆42Updated last month
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆39Updated 4 years ago
- Differentiable and numerically stable implementation of the matrix exponential☆33Updated 4 years ago
- SE(3) Equivariant Augmented Coupling Flows. NeurIPS 2023.☆24Updated last year
- Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"☆54Updated last year
- The Energy Transformer block, in JAX☆57Updated last year
- Official repository for the paper "Neural Differential Equations for Learning to Program Neural Nets Through Continuous Learning Rules" (…☆22Updated 2 years ago
- Code for the Thermodynamic Variational Objective☆26Updated 2 years ago
- A set of tests for evaluating large-scale algorithms for Wasserstein-2 transport maps computation (NeurIPS 2021)☆40Updated 2 years ago
- Riemannian Convex Potential Maps☆67Updated 2 years ago