mcbal / afemLinks
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
Sorting:
- Deterministic particle dynamics for simulating Fokker-Planck probability flows☆24Updated 2 years ago
- Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX☆41Updated last year
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 3 years ago
- Differentiable Euler Characteristic Transform☆17Updated last year
- Hierarchical variational models for physics.☆17Updated 5 years ago
- Implementation of Action Matching for the Schrödinger equation☆24Updated 2 years ago
- orbital MCMC☆10Updated 4 years ago
- A minimal implementation of a VAE with BinConcrete (relaxed Bernoulli) latent distribution in TensorFlow.☆22Updated 5 years ago
- ☆52Updated 2 years ago
- Riemannian Optimization Using JAX☆49Updated last year
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated last year
- Pytorch implement of the paper Neural Canonical Transformation with Symplectic Flows☆29Updated 5 years ago
- Library for normalizing flows and neural flows.☆24Updated 3 years ago
- ☆19Updated 3 years ago
- Course Website☆9Updated 3 years ago
- Implementation of deep implicit attention in PyTorch☆65Updated 3 years ago
- Implementation of Action Matching☆44Updated 2 years ago
- ☆38Updated 4 years ago
- "Parameter origami" -- folding and unfolding collections of parameters for optimization and sensitivity analysis.☆14Updated last year
- u-MPS implementation and experimentation code used in the paper Tensor Networks for Probabilistic Sequence Modeling (https://arxiv.org/ab…☆19Updated 4 years ago
- A generic interface for linear algebra backends☆73Updated 3 months ago
- This repository provides open-source code for sparse continuous distributions and corresponding Fenchel-Young losses.☆16Updated 2 years ago
- Code for the paper "Disentangling images with Lie group transformations and sparse coding" (2023). https://proceedings.mlr.press/v197/cha…☆13Updated 4 years ago
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆43Updated last year
- Bayesian Optimisation for String Spaces☆24Updated 3 years ago
- Implementation of various equivariant models in JAX☆12Updated last year
- PyTorch implementation for "Probabilistic Circuits for Variational Inference in Discrete Graphical Models", NeurIPS 2020☆17Updated 3 years ago
- Code for the paper "Randomly pivoted Cholesky: Practical approximation of a kernel matrix with few entry evaluations"☆29Updated 8 months ago
- An efficient method for the conversion from internal to Cartesian coordinates that utilizes the platform-agnostic JAX Python library.☆17Updated last year
- Pytorch code for Sampling in Combinatorial Spaces with SurVAE Flow Augmented MCMC☆12Updated 4 years ago