mcbal / spin-model-transformersLinks
Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX
☆43Updated last year
Alternatives and similar repositories for spin-model-transformers
Users that are interested in spin-model-transformers are comparing it to the libraries listed below
Sorting:
- Implementation of approximate free-energy minimization in PyTorch☆20Updated 4 years ago
- Exact OU processes with JAX☆56Updated 8 months ago
- Tools for building equivariant polynomials on reductive Lie groups.☆35Updated 2 years ago
- Deterministic particle dynamics for simulating Fokker-Planck probability flows☆26Updated 2 years ago
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆31Updated 2 years ago
- Code used by the "Clifford Group Equivariant Neural Networks" paper.☆85Updated last year
- ☆53Updated 3 years ago
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 4 years ago
- Implementation of Action Matching for the Schrödinger equation☆25Updated 2 years ago
- Implementation of Action Matching☆48Updated 2 years ago
- Implementation of Forward Laplacian algorithm in JAX☆89Updated last month
- Riemannian Optimization Using JAX☆53Updated 2 years ago
- diffusionjax is a simple and accessible diffusion models package in JAX☆47Updated 10 months ago
- Pytorch implement of the paper Neural Canonical Transformation with Symplectic Flows☆30Updated 5 years ago
- Implementation of various equivariant models in JAX☆12Updated last year
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]☆108Updated last month
- This repository contains the official code for Energy Transformer---an efficient Energy-based Transformer variant for graph classificatio…☆25Updated last year
- Repository for <Lorentz-Equivariant Geometric Algebra Transformers for High-Energy Physics> (J. Spinner et al 2024)☆29Updated 5 months ago
- Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.☆12Updated last year
- DeepErwin is a python 3.8+ package that implements and optimizes JAX 2.x wave function models for numerical solutions to the multi-electr…☆63Updated 7 months ago
- Clifford-Steerable Convolutional Neural Networks [ICML'24]☆49Updated 6 months ago
- Experiements on how conditional mutual information affects the performance of neural quantum states.☆13Updated last year
- A collection of graph neural networks implementations in JAX☆35Updated last year
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆137Updated last year
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆65Updated last year
- SE(3) Equivariant Augmented Coupling Flows. NeurIPS 2023.☆24Updated last year
- Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]☆29Updated 8 months ago
- This repository contains the data and scripts necessary to reproduce the results presented in the paper: **"Scalable molecular simulation…☆48Updated last year
- Library for normalizing flows and neural flows.☆25Updated 3 years ago
- Bayesian algorithm execution (BAX)☆53Updated 4 years ago