juho-lee / set_transformer

Related projects ⓘ

Alternatives and complementary repositories for set_transformer

• manzilzaheer / DeepSets
  ☆284Updated last year
• wgrathwohl / JEM
  Project site for "Your Classifier is Secretly an Energy-Based Model and You Should Treat it Like One"
  ☆417Updated 2 years ago
• jeanfeydy / geomloss
  Geometric loss functions between point clouds, images and volumes
  ☆595Updated 9 months ago
• geoopt / geoopt
  Riemannian Adaptive Optimization Methods with pytorch optim
  ☆850Updated 6 months ago
• rusty1s / pytorch_cluster
  PyTorch Extension Library of Optimized Graph Cluster Algorithms
  ☆823Updated 2 months ago
• nadavbh12 / VQ-VAE
  Minimalist implementation of VQ-VAE in Pytorch
  ☆512Updated 3 years ago
• martius-lab / blackbox-backprop
  Torch modules that wrap blackbox combinatorial solvers according to the method presented in "Differentiating Blackbox Combinatorial Solve…
  ☆337Updated 2 years ago
• locuslab / deq
  [NeurIPS'19] Deep Equilibrium Models
  ☆727Updated 2 years ago
• rusty1s / pytorch_sparse
  PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations
  ☆1,009Updated 2 months ago
• nicola-decao / s-vae-pytorch
  Pytorch implementation of Hyperspherical Variational Auto-Encoders
  ☆351Updated 4 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,589Updated 2 years ago
• YongfeiYan / Gumbel_Softmax_VAE
  PyTorch implementation of a Variational Autoencoder with Gumbel-Softmax Distribution
  ☆202Updated 6 years ago
• mfinzi / LieConv
  ☆258Updated last year
• lezcano / geotorch
  Constrained optimization toolkit for PyTorch
  ☆657Updated 2 years ago
• dfdazac / wassdistance
  Approximating Wasserstein distances with PyTorch
  ☆455Updated last year
• lucidrains / perceiver-pytorch
  Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch
  ☆1,092Updated last year
• rdevon / DIM
  Deep InfoMax (DIM), or "Learning Deep Representations by Mutual Information Estimation and Maximization"
  ☆806Updated 5 years ago
• rusty1s / pytorch_scatter
  PyTorch Extension Library of Optimized Scatter Operations
  ☆1,557Updated 2 months ago
• teddykoker / torchsort
  Fast, differentiable sorting and ranking in PyTorch
  ☆775Updated 10 months ago
• wjmaddox / swa_gaussian
  Code repo for "A Simple Baseline for Bayesian Uncertainty in Deep Learning"
  ☆451Updated last year
• Cyanogenoid / dspn
  [NeurIPS 2019] Deep Set Prediction Networks
  ☆100Updated 4 years ago
• deep-spin / entmax
  The entmax mapping and its loss, a family of sparse softmax alternatives.
  ☆415Updated 4 months ago
• gpeyre / SinkhornAutoDiff
  Toolbox to integrate optimal transport loss functions using automatic differentiation and Sinkhorn's algorithm
  ☆431Updated 6 years ago
• YannDubs / disentangling-vae
  Experiments for understanding disentanglement in VAE latent representations
  ☆795Updated last year
• JiaxuanYou / graph-generation
  GraphRNN: Generating Realistic Graphs with Deep Auto-regressive Models
  ☆696Updated 2 years ago
• lucidrains / linear-attention-transformer
  Transformer based on a variant of attention that is linear complexity in respect to sequence length
  ☆695Updated 6 months ago
• ssnl / align_uniform
  Open source code for paper "Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hypersphere" ICML 2…
  ☆432Updated 2 years ago
• matthewvowels1 / Awesome-VAEs
  A curated list of awesome work on VAEs, disentanglement, representation learning, and generative models.
  ☆786Updated 3 years ago
• ikostrikov / pytorch-flows
  PyTorch implementations of algorithms for density estimation
  ☆576Updated 3 years ago
• chrischute / real-nvp
  Implementation of Real NVP in PyTorch
  ☆224Updated 5 years ago