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
- Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX☆38Updated last year
- Deterministic particle dynamics for simulating Fokker-Planck probability flows☆24Updated 2 years ago
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 3 years ago
- Exact OU processes with JAX☆42Updated 2 weeks ago
- Riemannian Optimization Using JAX☆48Updated last year
- ☆34Updated 2 years ago
- Implementation of Action Matching for the Schrödinger equation☆24Updated last year
- orbital MCMC☆10Updated 3 years ago
- Implementation of deep implicit attention in PyTorch☆65Updated 3 years ago
- Differentiable Euler Characteristic Transform☆17Updated 9 months ago
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- An implementation of ESM2 in Equinox+JAX☆25Updated last month
- PyTorch implementation for "Probabilistic Circuits for Variational Inference in Discrete Graphical Models", NeurIPS 2020☆17Updated 3 years ago
- Pytorch code for Sampling in Combinatorial Spaces with SurVAE Flow Augmented MCMC☆12Updated 4 years ago
- Code for NeurIPS 2024 paper: "Noether's razor: Learning Conserved Quantities" by Tycho F. A. van der Ouderaa, Mark van der Wilk, Pim de H…☆10Updated 5 months ago
- Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"☆54Updated last year
- ☆50Updated 2 years ago
- Implementation of Action Matching☆40Updated last year
- ☆19Updated 3 years ago
- Hierarchical variational models for physics.☆17Updated 4 years ago
- Course Website☆9Updated 3 years ago
- Bayesian Optimisation for String Spaces☆24Updated 3 years ago
- JAXChem is a JAX-based deep learning library for complex and versatile chemical modeling☆79Updated 4 years ago
- Implementation of various equivariant models in JAX☆12Updated 11 months ago
- "Parameter origami" -- folding and unfolding collections of parameters for optimization and sensitivity analysis.☆14Updated last year
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆39Updated 4 years ago
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆56Updated last year
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated last year
- SE(3) Equivariant Augmented Coupling Flows. NeurIPS 2023.☆24Updated last year
- Code for the paper "Randomly pivoted Cholesky: Practical approximation of a kernel matrix with few entry evaluations"☆28Updated 5 months ago