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
- Riemannian Optimization Using JAX☆49Updated last year
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 3 years ago
- orbital MCMC☆10Updated 3 years ago
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated last year
- Differentiable Euler Characteristic Transform☆17Updated 11 months ago
- Implementation of Action Matching for the Schrödinger equation☆24Updated last year
- Implementation of Action Matching☆44Updated 2 years ago
- A collection of graph neural networks implementations in JAX☆32Updated last year
- Library for normalizing flows and neural flows.☆24Updated 3 years ago
- ☆52Updated 2 years ago
- An implementation of ESM2 in Equinox+JAX☆26Updated 3 months ago
- Hierarchical variational models for physics.☆17Updated 5 years ago
- Implementation of deep implicit attention in PyTorch☆65Updated 3 years ago
- Exact OU processes with JAX☆48Updated 2 months ago
- Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"☆54Updated last year
- ☆51Updated last month
- A minimal implementation of a VAE with BinConcrete (relaxed Bernoulli) latent distribution in TensorFlow.☆22Updated 5 years ago
- Course Website☆9Updated 3 years ago
- Implementation of various equivariant models in JAX☆12Updated last year
- Pytorch implement of the paper Neural Canonical Transformation with Symplectic Flows☆29Updated 5 years ago
- This repository contains PyTorch implementations of various random feature maps for dot product kernels.☆21Updated 10 months ago
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆30Updated last year
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆43Updated last year
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- ☆32Updated 8 months ago
- Bayesian Optimisation for String Spaces☆24Updated 3 years ago
- Neurally-derived Potts models for protein design, inspired by dTERMen☆14Updated 2 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆39Updated 4 years ago