Alternatives and similar repositories for set_transformer

Users that are interested in set_transformer are comparing it to the libraries listed below

• manzilzaheer / DeepSets
  ☆291Updated 2 years ago
• nicola-decao / s-vae-pytorch
  Pytorch implementation of Hyperspherical Variational Auto-Encoders
  ☆363Updated 5 years ago
• jeanfeydy / geomloss
  Geometric loss functions between point clouds, images and volumes
  ☆624Updated last month
• rusty1s / pytorch_cluster
  PyTorch Extension Library of Optimized Graph Cluster Algorithms
  ☆869Updated 3 weeks ago
• ssnl / align_uniform
  Open source code for paper "Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hypersphere" ICML 2…
  ☆442Updated 2 years ago
• rusty1s / pytorch_sparse
  PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations
  ☆1,059Updated last month
• lucidrains / performer-pytorch
  An implementation of Performer, a linear attention-based transformer, in Pytorch
  ☆1,131Updated 3 years ago
• lucidrains / sinkhorn-transformer
  Sinkhorn Transformer - Practical implementation of Sparse Sinkhorn Attention
  ☆262Updated 3 years ago
• locuslab / deq
  [NeurIPS'19] Deep Equilibrium Models
  ☆748Updated 2 years ago
• google-research / fast-soft-sort
  Fast Differentiable Sorting and Ranking
  ☆603Updated last year
• teddykoker / torchsort
  Fast, differentiable sorting and ranking in PyTorch
  ☆813Updated this week
• geoopt / geoopt
  Riemannian Adaptive Optimization Methods with pytorch optim
  ☆923Updated last year
• wgrathwohl / JEM
  Project site for "Your Classifier is Secretly an Energy-Based Model and You Should Treat it Like One"
  ☆423Updated 2 years ago
• lucidrains / perceiver-pytorch
  Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch
  ☆1,142Updated last year
• deep-spin / entmax
  The entmax mapping and its loss, a family of sparse softmax alternatives.
  ☆436Updated 10 months ago
• rusty1s / pytorch_scatter
  PyTorch Extension Library of Optimized Scatter Operations
  ☆1,639Updated 3 weeks ago
• YongfeiYan / Gumbel_Softmax_VAE
  PyTorch implementation of a Variational Autoencoder with Gumbel-Softmax Distribution
  ☆208Updated 6 years ago
• EmilienDupont / neural-processes
  Pytorch implementation of Neural Processes for functions and images
  ☆230Updated 3 years ago
• martius-lab / blackbox-backprop
  Torch modules that wrap blackbox combinatorial solvers according to the method presented in "Differentiating Blackbox Combinatorial Solve…
  ☆343Updated 3 years ago
• emilemathieu / pvae
  code for "Continuous Hierarchical Representations with Poincaré Variational Auto-Encoders".
  ☆124Updated last year
• dfdazac / wassdistance
  Approximating Wasserstein distances with PyTorch
  ☆459Updated 2 years ago
• haofuml / cyclical_annealing
  ☆193Updated 2 years ago
• YannDubs / disentangling-vae
  Experiments for understanding disentanglement in VAE latent representations
  ☆815Updated 2 years ago
• idiap / fast-transformers
  Pytorch library for fast transformer implementations
  ☆1,703Updated 2 years ago
• mfinzi / LieConv
  ☆272Updated 2 years ago
• WeiChengTseng / Pytorch-PCGrad
  Pytorch reimplementation for "Gradient Surgery for Multi-Task Learning"
  ☆341Updated 3 years ago
• facebookresearch / higher
  higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…
  ☆1,618Updated 3 years ago
• tkipf / c-swm
  Contrastive Learning of Structured World Models
  ☆392Updated 4 years ago
• lucidrains / graph-transformer-pytorch
  Implementation of Graph Transformer in Pytorch, for potential use in replicating Alphafold2
  ☆220Updated last year
• KrisKorrel / sparsemax-pytorch
  Implementation of Sparsemax activation in Pytorch
  ☆160Updated 4 years ago