Alternatives and similar repositories for jax-gat

Users that are interested in jax-gat are comparing it to the libraries listed below

• lucidrains / ESBN-pytorch
  Usable implementation of Emerging Symbol Binding Network (ESBN), in Pytorch
  ☆25Updated 4 years ago
• IndustAI / learning-group-structure
  Code associated with our paper "Learning Group Structure and Disentangled Representations of Dynamical Environments"
  ☆15Updated 2 years ago
• juliagusak / neural-ode-metasolver
  Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561
  ☆25Updated 4 years ago
• Holmeswww / PPOGAN
  ☆25Updated last year
• stanfordmlgroup / disentanglement
  Official repository for our ICLR 2021 paper Evaluating the Disentanglement of Deep Generative Models with Manifold Topology
  ☆36Updated 4 years ago
• lucidrains / deep-linear-network
  A simple implementation of a deep linear Pytorch module
  ☆21Updated 5 years ago
• lucidrains / isab-pytorch
  An implementation of (Induced) Set Attention Block, from the Set Transformers paper
  ☆64Updated 2 years ago
• joeybose / HyperbolicNF
  ICML 2020 Paper: Latent Variable Modelling with Hyperbolic Normalizing Flows
  ☆54Updated 2 years ago
• lucidrains / lie-transformer-pytorch
  Implementation of Lie Transformer, Equivariant Self-Attention, in Pytorch
  ☆96Updated 4 years ago
• AranKomat / Metroplex
  ☆21Updated 2 years ago
• j-towns / vdvae-jax
  Very deep VAEs in JAX/Flax
  ☆46Updated 4 years ago
• IDSIA / recurrent-fwp
  Official repository for the paper "Going Beyond Linear Transformers with Recurrent Fast Weight Programmers" (NeurIPS 2021)
  ☆50Updated 4 months ago
• ZhengyaoJiang / GTG
  Source code of "Grid-to-Graph: Flexible Spatial Relational Inductive Biases for Reinforcement Learning" (AAMAS 2021).
  ☆28Updated 4 years ago
• facebookresearch / Addressing-the-Topological-Defects-of-Disentanglement
  Repo reproducing experimental results in "Addressing the Topological Defects of Disentanglement"
  ☆22Updated 3 years ago
• lucidrains / mlp-gpt-jax
  A GPT, made only of MLPs, in Jax
  ☆58Updated 4 years ago
• nec-research / tf-imle
  Tensorflow implementation and notebooks for Implicit Maximum Likelihood Estimation
  ☆67Updated 3 years ago
• RedRyan111 / GLOM
  An implementation of 2021 paper by Geoffrey Hinton: "How to represent part-whole hierarchies in a neural network" in Pytorch.
  ☆57Updated 4 years ago
• matbun / EBM--Generative-Energy-Based-Modeling
  Energy Based Models are a quite novel technique for density estimation. In this university project I explore this new research topic and …
  ☆16Updated 4 years ago
• lucidrains / all-normalization-transformer
  A simple Transformer where the softmax has been replaced with normalization
  ☆20Updated 5 years ago
• PAL-ML / PEARL_v1
  ☆30Updated 3 years ago
• antofuller / configaformers
  A python library for highly configurable transformers - easing model architecture search and experimentation.
  ☆49Updated 3 years ago
• choidami / sst
  ☆50Updated 5 years ago
• lucidrains / remixer-pytorch
  Implementation of the Remixer Block from the Remixer paper, in Pytorch
  ☆36Updated 4 years ago
• ColinQiyangLi / AdaCat
  AdaCat
  ☆49Updated 3 years ago
• mcbal / deep-implicit-attention
  Implementation of deep implicit attention in PyTorch
  ☆65Updated 4 years ago
• shwinshaker / LipGrow
  An adaptive training algorithm for residual network
  ☆17Updated 5 years ago
• lucidrains / local-attention-flax
  Local Attention - Flax module for Jax
  ☆22Updated 4 years ago
• lucidrains / triangle-multiplicative-module
  Implementation of the Triangle Multiplicative module, used in Alphafold2 as an efficient way to mix rows or columns of a 2d feature map, …
  ☆36Updated 4 years ago
• POSTECH-CVLab / Brick-by-Brick
  Official repository of Brick-by-Brick, presented at NeurIPS-2021
  ☆15Updated 3 years ago
• joelouismarino / variational_rl
  Variational Reinforcement Learning
  ☆16Updated last year