gengala / egnca
Official implementation of E(n)-equivariant Graph Neural Cellular Automata
☆25Updated 9 months ago
Alternatives and similar repositories for egnca:
Users that are interested in egnca are comparing it to the libraries listed below
- Code for "Invariance Learning in Deep Neural Networks with Differentiable Laplace Approximations"☆23Updated 2 years ago
- Automatic Integration for Neural Spatio-Temporal Point Process models (AI-STPP) is a new paradigm for exact, efficient, non-parametric inf…☆24Updated 4 months ago
- ☆13Updated last year
- Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)☆27Updated 2 years ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- The Energy Transformer block, in JAX☆56Updated last year
- ☆26Updated 2 years ago
- ☆24Updated last year
- Code repo for ICLR 24 BlogPost titled "Building Diffusion Model's theory from ground up"☆16Updated last year
- Scalable Computation of Hessian Diagonals☆13Updated 8 months ago
- JAX implementation of Graph Attention Networks☆13Updated 3 years ago
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"☆31Updated last year
- Variational Reinforcement Learning☆16Updated 6 months ago
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆19Updated 4 years ago
- Code for minimum-entropy coupling.☆31Updated 7 months ago
- Deep Networks Grok All the Time and Here is Why☆27Updated 9 months ago
- ☆37Updated 2 years ago
- Official repository for the paper "Neural Differential Equations for Learning to Program Neural Nets Through Continuous Learning Rules" (…☆19Updated 2 years ago
- Generalised UDRL☆37Updated 2 years ago
- Contact-Aware Symplectic Integrator Network☆11Updated last year
- Variational Autoencoder with non-euclidean (hyperbolic) latent space☆11Updated 2 years ago
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆42Updated last year
- Riemannian Convex Potential Maps☆67Updated last year
- A minimal working example of Free-Form Jacobian of Reversible Dynamics☆13Updated 5 years ago
- Neural Graphical models are neural network based graphical models that offer richer representation, faster inference & sampling☆28Updated last year
- code associated with paper "Sparse Bayesian Optimization"☆26Updated last year
- Repo for paper: Exploring the Power of Graph Neural Networks in Solving Linear Optimization Problems, accepted at AISTATS 2024☆12Updated last year
- A PyTorch implementation of a Generative Flow Network (GFlowNet) proposed by Bengio et al. (2021)☆42Updated last year
- Neural Optimal Transport with Lagrangian Costs☆52Updated 7 months ago
- ☆31Updated 10 months ago