gcucurull / jax-gat
JAX implementation of Graph Attention Networks
☆13Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for jax-gat
- Code associated with our paper "Learning Group Structure and Disentangled Representations of Dynamical Environments"☆15Updated last year
- Unofficially Implements https://arxiv.org/abs/2112.05682 to get Linear Memory Cost on Attention for PyTorch☆12Updated 2 years ago
- Graph Convolutional Networks in JAX☆31Updated 3 years ago
- Variational Reinforcement Learning☆16Updated 3 months ago
- Usable implementation of Emerging Symbol Binding Network (ESBN), in Pytorch☆23Updated 3 years ago
- Energy Based Models are a quite novel technique for density estimation. In this university project I explore this new research topic and …☆15Updated 3 years ago
- Optimization and Generalization Analysis of Transduction through Gradient Boosting and Application to Multi-scale Graph Neural Networks☆13Updated 4 years ago
- A framework for implementing equivariant DL☆10Updated 3 years ago
- Official repository for the paper "Going Beyond Linear Transformers with Recurrent Fast Weight Programmers" (NeurIPS 2021)☆47Updated last year
- Perceiver (transformer variant) implemented in JAX and Flax☆11Updated 3 years ago
- ☆36Updated 2 years ago
- Implementation of the Triangle Multiplicative module, used in Alphafold2 as an efficient way to mix rows or columns of a 2d feature map, …☆29Updated 3 years ago
- ☆21Updated last year
- The implementation of "The Kanerva Machine" with Pytorch and Pyro☆12Updated 6 years ago
- Automatically generate simple meta-learning tasks from a very large space☆15Updated last year
- Clockwork VAEs in JAX/Flax☆32Updated 3 years ago
- ☆29Updated 2 years ago
- ☆24Updated 6 months ago
- Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)☆27Updated 2 years ago
- ☆21Updated 2 years ago
- A basic implementation of the paper Eigengame : PCA as a Nash Equilibrium☆21Updated 3 years ago
- Implementation of Tranception, an attention network, paired with retrieval, that is SOTA for protein fitness prediction☆31Updated 2 years ago
- ☆16Updated last year
- Contact-Aware Symplectic Integrator Network☆11Updated last year
- notebooks of cool EBM visualizations☆16Updated 3 years ago
- Simple, extensible implementations of some meta-learning algorithms in Jax☆10Updated 4 years ago
- Official repository for our ICLR 2021 paper Evaluating the Disentanglement of Deep Generative Models with Manifold Topology☆36Updated 3 years ago
- ☆12Updated 3 years ago