gcucurull / jax-gcnLinks
Graph Convolutional Networks in JAX
☆32Updated 4 years ago
Alternatives and similar repositories for jax-gcn
Users that are interested in jax-gcn are comparing it to the libraries listed below
Sorting:
- JAX implementation of Graph Attention Networks☆13Updated 3 years ago
- JAXChem is a JAX-based deep learning library for complex and versatile chemical modeling☆79Updated 4 years ago
- Repository for Autobahn: Automorphism Based Graph Neural Networks☆29Updated 3 years ago
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- Lie Algebra Convolutional Network implementation☆44Updated 3 years ago
- Library for normalizing flows and neural flows.☆24Updated 3 years ago
- Code repository for "Finding symmetry breaking order parameters with Euclidean Neural Networks"☆14Updated 4 years ago
- ☆15Updated 4 years ago
- Deterministic particle dynamics for simulating Fokker-Planck probability flows☆24Updated 2 years ago
- Pytorch implement of the paper Neural Canonical Transformation with Symplectic Flows☆29Updated 5 years ago
- Steerable E(3) GNN in jax☆24Updated last year
- orbital MCMC☆10Updated 3 years ago
- ☆51Updated 2 years ago
- JAX implementation of Learning to learn by gradient descent by gradient descent☆27Updated 7 months ago
- Implementation of Geometric Vector Perceptron, a simple circuit for 3d rotation equivariance for learning over large biomolecules, in Pyt…☆74Updated 3 years ago
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 3 years ago
- Ab-initio simulation of interacting fermions with equivariant normalizing flow.☆34Updated 2 years ago
- Numerically Solving Parametric Families of High-Dimensional Kolmogorov Partial Differential Equations via Deep Learning (NeurIPS 2020)☆22Updated 2 years ago
- Reference implementation of "Ab-Initio Potential Energy Surfaces by Pairing GNNs with Neural Wave Functions" (ICLR, 2022) and "Sampling-f…☆31Updated 11 months ago
- Pytorch code for Sampling in Combinatorial Spaces with SurVAE Flow Augmented MCMC☆12Updated 4 years ago
- Combining smooth constraint for building DAG with normalizing flow in order to replace autoregressive transformations while keeping tract…☆45Updated last year
- Pytorch reimplementation of Molecule Attention Transformer, which uses a transformer to tackle the graph-like structure of molecules☆59Updated 4 years ago
- A short and easy PyTorch implementation of E(n) Equivariant Graph Neural Networks☆131Updated 3 years ago
- Implementation of approximate free-energy minimization in PyTorch☆19Updated 3 years ago
- ☆38Updated 2 years ago
- Canonical normalizing flows☆10Updated 6 years ago
- PyTorch block-diagonal ODE CUDA solver, designed for gradient-based optimization☆15Updated 5 years ago
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"☆54Updated last year
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated last year