Alternatives and similar repositories for pmae

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

• mkirchhof / url
  Uncertainty-aware representation learning (URL) benchmark
  ☆106Updated 11 months ago
• facebookresearch / RCDM
  Visualizing representations with diffusion based conditional generative model.
  ☆106Updated 2 years ago
• lucidrains / hl-gauss-pytorch
  The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch
  ☆73Updated 2 months ago
• samar-khanna / ExPLoRA
  Official code repository for ICML 2025 paper: "ExPLoRA: Parameter-Efficient Extended Pre-training to Adapt Vision Transformers under Doma…
  ☆51Updated last month
• AhmedImtiazPrio / grok-adversarial
  Deep Networks Grok All the Time and Here is Why
  ☆38Updated last year
• s-sahoo / MuLAN
  [NeurIPS 2024, spotlight] Multivariate Learned Adaptive Noise for Diffusion Models
  ☆31Updated last year
• ShadeAlsha / ICon
  ICLR 2025 - official implementation for "I-Con: A Unifying Framework for Representation Learning"
  ☆127Updated this week
• ethanbar11 / ssm_2d
  More dimensions = More fun
  ☆26Updated last year
• nreHieW / minARImageGen
  Autoregressive Image Generation
  ☆31Updated 7 months ago
• james-oldfield / muMoE
  [NeurIPS'24] Multilinear Mixture of Experts: Scalable Expert Specialization through Factorization
  ☆39Updated last year
• ml-jku / MIM-Refiner
  A Contrastive Learning Boost from Intermediate Pre-Trained Representations
  ☆43Updated last year
• WalBouss / LeGrad
  [ICCV25] Official Implementation of LeGrad
  ☆87Updated last year
• alhojel / visual_task_vectors
  ☆40Updated last year
• TomerRonen34 / mixed-resolution-vit
  ☆56Updated 2 years ago
• nickjiang2378 / test-time-registers
  [NeurIPS '25 Spotlight] Official Pytorch implementation of "Vision Transformers Don't Need Trained Registers"
  ☆172Updated 4 months ago
• AvivNavon / DWSNets
  Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces [ICML 2023]
  ☆90Updated 2 years ago
• TonyLianLong / CrossMAE
  Official Implementation of the CrossMAE paper: Rethinking Patch Dependence for Masked Autoencoders
  ☆131Updated 10 months ago
• lucidrains / gradnorm-pytorch
  A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch
  ☆126Updated 5 months ago
• LeslieTrue / CPP
  This is the official implementation for Image Clustering via the Principle of Rate Reduction in the Age of Pretrained Models.
  ☆32Updated 2 years ago
• lucidrains / multimodal-dit-pytorch
  Implementation of a multimodal diffusion transformer in Pytorch
  ☆107Updated last year
• davidpicard / HoMM
  High order Moment Models
  ☆42Updated 2 months ago
• njwfish / DistributionEmbeddings
  ☆38Updated 3 months ago
• lucidrains / spline-based-transformer
  Implementation of the proposed Spline-Based Transformer from Disney Research
  ☆105Updated last year
• badripatro / mamba360
  State Space Models
  ☆72Updated last year
• goombalab / hydra
  Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"
  ☆170Updated last year
• giannisdaras / ambient-tweedie
  [ICML 2024]: Official implementation for the paper: "Consistent Diffusion Meets Tweedie"
  ☆53Updated last year
• oripress / EntropyEnigma
  Official code for the ICML 2024 paper "The Entropy Enigma: Success and Failure of Entropy Minimization"
  ☆55Updated last year
• lucidrains / deep-cross-attention
  Implementation of the proposed DeepCrossAttention by Heddes et al at Google research, in Pytorch
  ☆96Updated 11 months ago
• GenRobo / MatMamba
  Code and pretrained models for the paper: "MatMamba: A Matryoshka State Space Model"
  ☆62Updated last year
• lucidrains / AMIE-pytorch
  Implementation of the general framework for AMIE, from the paper "Towards Conversational Diagnostic AI", out of Google Deepmind
  ☆74Updated last year