Alternatives and similar repositories for pmae

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

• AhmedImtiazPrio / grok-adversarial
  Deep Networks Grok All the Time and Here is Why
  ☆37Updated last year
• mkirchhof / url
  Uncertainty-aware representation learning (URL) benchmark
  ☆105Updated 6 months ago
• facebookresearch / RCDM
  Visualizing representations with diffusion based conditional generative model.
  ☆100Updated 2 years ago
• nreHieW / minARImageGen
  Autoregressive Image Generation
  ☆32Updated 3 months ago
• ethanbar11 / ssm_2d
  More dimensions = More fun
  ☆25Updated last year
• WalBouss / LeGrad
  [ICCV25] Official Implementation of LeGrad
  ☆78Updated 11 months ago
• hyperevolnet / Terminator
  The official repository for HyperZ⋅Z⋅W Operator Connects Slow-Fast Networks for Full Context Interaction.
  ☆39Updated 5 months ago
• EPFL-VILAB / fm-vision-evals
  ☆71Updated 2 months ago
• SamsungSAILMontreal / PAPA
  Repository for the PopulAtion Parameter Averaging (PAPA) paper
  ☆27Updated last year
• lucidrains / hl-gauss-pytorch
  The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch
  ☆65Updated last month
• gregorbachmann / scaling_mlps
  ☆51Updated last year
• giannisdaras / ambient-tweedie
  [ICML 2024]: Official implementation for the paper: "Consistent Diffusion Meets Tweedie"
  ☆53Updated last year
• TomerRonen34 / mixed-resolution-vit
  ☆52Updated last year
• SriramB-98 / vit-decompose
  ☆23Updated 8 months ago
• james-oldfield / muMoE
  [NeurIPS'24] Multilinear Mixture of Experts: Scalable Expert Specialization through Factorization
  ☆34Updated 11 months ago
• davidpicard / HoMM
  High order Moment Models
  ☆42Updated last week
• AvivNavon / DWSNets
  Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]
  ☆89Updated 2 years ago
• alhojel / visual_task_vectors
  ☆39Updated last year
• ShadeAlsha / ICon
  ICLR 2025 - official implementation for "I-Con: A Unifying Framework for Representation Learning"
  ☆112Updated 2 months ago
• amirbar / StoP
  ☆11Updated last year
• layer6ai-labs / calo-forest
  A scalable implementation of diffusion and flow-matching with XGBoost models, applied to calorimeter data.
  ☆18Updated 10 months ago
• s-sahoo / MuLAN
  [NeurIPS 2024, spotlight] Multivariate Learned Adaptive Noise for Diffusion Models
  ☆30Updated 9 months ago
• njwfish / DistributionEmbeddings
  ☆30Updated last month
• lucidrains / gradnorm-pytorch
  A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch
  ☆107Updated 3 weeks ago
• ChenWu98 / algorithmic-creativity
  [ICML 2025] Roll the dice & look before you leap: Going beyond the creative limits of next-token prediction
  ☆68Updated 3 months ago
• lucidrains / spline-based-transformer
  Implementation of the proposed Spline-Based Transformer from Disney Research
  ☆103Updated 10 months ago
• google-deepmind / tips
  ☆103Updated 5 months ago
• willGuimont / learnable_fourier_positional_encoding
  Learnable Fourier Features for Multi-Dimensional Spatial Positional Encoding
  ☆56Updated 11 months ago
• tae898 / vae-diffusion
  ☆32Updated 2 months ago
• lucidrains / deep-cross-attention
  Implementation of the proposed DeepCrossAttention by Heddes et al at Google research, in Pytorch
  ☆92Updated 6 months ago