Alternatives and similar repositories for jax-gat:

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

• LCS2-IIITD / TransEvolve
  ☆11Updated 3 years ago
• IndustAI / learning-group-structure
  Code associated with our paper "Learning Group Structure and Disentangled Representations of Dynamical Environments"
  ☆15Updated 2 years ago
• lucidrains / ESBN-pytorch
  Usable implementation of Emerging Symbol Binding Network (ESBN), in Pytorch
  ☆24Updated 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
• CHARM-Tx / linear_mem_attention_pytorch
  Unofficially Implements https://arxiv.org/abs/2112.05682 to get Linear Memory Cost on Attention for PyTorch
  ☆12Updated 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
• AranKomat / Metroplex
  ☆21Updated 2 years ago
• IDSIA / recurrent-fwp
  Official repository for the paper "Going Beyond Linear Transformers with Recurrent Fast Weight Programmers" (NeurIPS 2021)
  ☆48Updated 2 years ago
• Holmeswww / PPOGAN
  ☆24Updated last year
• joelouismarino / variational_rl
  Variational Reinforcement Learning
  ☆16Updated 9 months ago
• ZhengyaoJiang / GTG
  Source code of "Grid-to-Graph: Flexible Spatial Relational Inductive Biases for Reinforcement Learning" (AAMAS 2021).
  ☆28Updated 3 years ago
• EleutherAI / equivariance
  A framework for implementing equivariant DL
  ☆10Updated 3 years ago
• remilepriol / causal-adaptation-speed
  Investigate the speed of adaptation of structural causal models
  ☆16Updated 4 years ago
• facebookresearch / neural-scs
  Neural Fixed-Point Acceleration for Convex Optimization
  ☆29Updated 2 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 3 years ago
• gwgundersen / rflvm
  Random feature latent variable models in Python
  ☆22Updated last year
• imd-iclr / imd
  The Shape of Data: Intrinsic Distance for Comparing Data Distributions
  ☆12Updated 5 years ago
• facebookresearch / Addressing-the-Topological-Defects-of-Disentanglement
  Repo reproducing experimental results in "Addressing the Topological Defects of Disentanglement"
  ☆22Updated 2 years ago
• petrapoklukar / GeomCA
  Official Python implementation of Geometric Component Analysis (GeomCA) algorithm.
  ☆9Updated 2 years ago
• Abhishaike / HyperProtoNetReproduce
  NeurIPS 2019 Paper Implementation
  ☆12Updated 2 years ago
• delta2323 / GB-GNN
  Optimization and Generalization Analysis of Transduction through Gradient Boosting and Application to Multi-scale Graph Neural Networks
  ☆13Updated 4 years ago
• spitis / deepnorms
  Code for An Inductive Bias for Distances: Neural Nets that Respect the Triangle Inequality (ICLR 2020)
  ☆11Updated 2 years ago
• facebookresearch / grounding-inductive-biases
  reproduces experiments from "Grounding inductive biases in natural images: invariance stems from variations in data"
  ☆17Updated 7 months ago
• Kajiyu / kanerva_machine
  The implementation of "The Kanerva Machine" with Pytorch and Pyro
  ☆12Updated 6 years ago
• arrigonialberto86 / set_transformer
  A TensorFlow implementation of the paper 'Set Transformer: A Framework for Attention-based Permutation-Invariant Neural Networks'
  ☆31Updated last year
• juliuskunze / cwvae-jax
  Clockwork VAEs in JAX/Flax
  ☆32Updated 3 years ago
• libeanim / contact-symplectic-integrator-network
  Contact-Aware Symplectic Integrator Network
  ☆14Updated 2 years ago
• nkeriven / random-graph-gnn
  ☆12Updated 3 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, …
  ☆29Updated 3 years ago
• bergen / EdgeTransformer
  ☆22Updated 3 years ago