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 5 years ago
• IndustAI / learning-group-structure
  Code associated with our paper "Learning Group Structure and Disentangled Representations of Dynamical Environments"
  ☆15Updated 3 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
• stanfordmlgroup / disentanglement
  Official repository for our ICLR 2021 paper Evaluating the Disentanglement of Deep Generative Models with Manifold Topology
  ☆36Updated 4 years ago
• ZhengyaoJiang / GTG
  Source code of "Grid-to-Graph: Flexible Spatial Relational Inductive Biases for Reinforcement Learning" (AAMAS 2021).
  ☆28Updated 4 years ago
• IDSIA / recurrent-fwp
  Official repository for the paper "Going Beyond Linear Transformers with Recurrent Fast Weight Programmers" (NeurIPS 2021)
  ☆51Updated 7 months ago
• Holmeswww / PPOGAN
  ☆25Updated last year
• lucidrains / isab-pytorch
  An implementation of (Induced) Set Attention Block, from the Set Transformers paper
  ☆66Updated 3 years ago
• CHARM-Tx / linear_mem_attention_pytorch
  Unofficially Implements https://arxiv.org/abs/2112.05682 to get Linear Memory Cost on Attention for PyTorch
  ☆12Updated 4 years ago
• POSTECH-CVLab / Brick-by-Brick
  Official repository of Brick-by-Brick, presented at NeurIPS-2021
  ☆15Updated 3 years ago
• lucidrains / deep-linear-network
  A simple implementation of a deep linear Pytorch module
  ☆21Updated 5 years ago
• AranKomat / Metroplex
  ☆21Updated 2 years ago
• choidami / sst
  ☆50Updated 5 years ago
• joeybose / HyperbolicNF
  ICML 2020 Paper: Latent Variable Modelling with Hyperbolic Normalizing Flows
  ☆54Updated 3 years ago
• attentionneuron / attentionneuron.github.io
  ☆26Updated 3 years ago
• ColinQiyangLi / AdaCat
  AdaCat
  ☆49Updated 3 years ago
• antofuller / configaformers
  A python library for highly configurable transformers - easing model architecture search and experimentation.
  ☆49Updated 4 years ago
• lucidrains / lie-transformer-pytorch
  Implementation of Lie Transformer, Equivariant Self-Attention, in Pytorch
  ☆97Updated 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
• j-towns / vdvae-jax
  Very deep VAEs in JAX/Flax
  ☆46Updated 4 years ago
• facebookresearch / neural-scs
  Neural Fixed-Point Acceleration for Convex Optimization
  ☆29Updated 3 years ago
• insilicomedicine / DD-VAE
  Deterministic Decoding for Discrete Data in Variational Autoencoders
  ☆24Updated 5 years ago
• mcbal / deep-implicit-attention
  Implementation of deep implicit attention in PyTorch
  ☆65Updated 4 years ago
• n2cholas / progan-flax
  Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation
  ☆12Updated 4 years ago
• joelouismarino / variational_rl
  Variational Reinforcement Learning
  ☆17Updated last year
• nec-research / tf-imle
  Tensorflow implementation and notebooks for Implicit Maximum Likelihood Estimation
  ☆67Updated 3 years ago
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆39Updated 3 years ago
• PAL-ML / PEARL_v1
  ☆30Updated 4 years ago
• gauravdhama / eigengame_deepmind
  A basic implementation of the paper Eigengame : PCA as a Nash Equilibrium
  ☆21Updated 4 years ago
• lucidrains / mlp-gpt-jax
  A GPT, made only of MLPs, in Jax
  ☆59Updated 4 years ago