mcbal / afem
Implementation of approximate free-energy minimization in PyTorch
☆18Updated 3 years ago
Related projects ⓘ
Alternatives and complementary repositories for afem
- Deterministic particle dynamics for simulating Fokker-Planck probability flows☆24Updated last year
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 3 years ago
- Library for normalizing flows and neural flows.☆24Updated 2 years ago
- Implementation of deep implicit attention in PyTorch☆63Updated 3 years ago
- Riemannian Optimization Using JAX☆45Updated last year
- ☆49Updated last year
- Implementation of Action Matching for the Schrödinger equation☆23Updated last year
- Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX☆32Updated 11 months ago
- Differentiable Euler Characteristic Transform☆17Updated 5 months ago
- PyTorch implementation for "Probabilistic Circuits for Variational Inference in Discrete Graphical Models", NeurIPS 2020☆15Updated 3 years ago
- Pytorch code for Sampling in Combinatorial Spaces with SurVAE Flow Augmented MCMC☆12Updated 3 years ago
- Bayesian Optimisation for String Spaces☆24Updated 3 years ago
- A minimal implementation of a VAE with BinConcrete (relaxed Bernoulli) latent distribution in TensorFlow.☆21Updated 4 years ago
- orbital MCMC☆10Updated 3 years ago
- SE(3) Equivariant Augmented Coupling Flows. NeurIPS 2023.☆23Updated 8 months ago
- Implementation of Action Matching☆36Updated last year
- Euclidean Wasserstein-2 optimal transportation☆42Updated last year
- Dive into Jax, Flax, XLA and C++☆31Updated 4 years ago
- This repository contains PyTorch implementations of various random feature maps for dot product kernels.☆18Updated 4 months ago
- ☆31Updated 2 years ago
- Stochastic Normalizing Flows☆73Updated 2 years ago
- DISCS: The code base for the Benchmark for Discrete Sampling☆46Updated 3 months ago
- "Parameter origami" -- folding and unfolding collections of parameters for optimization and sensitivity analysis.☆13Updated 10 months ago
- ☆36Updated 4 years ago
- Composable kernels for scikit-learn implemented in JAX.☆42Updated 4 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆38Updated 4 years ago
- Combining smooth constraint for building DAG with normalizing flow in order to replace autoregressive transformations while keeping tract…☆44Updated last year
- Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"☆52Updated last year
- A PyTorch implementation of a Generative Flow Network (GFlowNet) proposed by Bengio et al. (2021)☆40Updated last year
- Course Website☆9Updated 2 years ago