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
  ☆105Updated 9 months 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
  ☆68Updated 3 weeks ago
• WalBouss / LeGrad
  [ICCV25] Official Implementation of LeGrad
  ☆83Updated last year
• nreHieW / minARImageGen
  Autoregressive Image Generation
  ☆31Updated 6 months ago
• facebookresearch / RCDM
  Visualizing representations with diffusion based conditional generative model.
  ☆103Updated 2 years ago
• njwfish / DistributionEmbeddings
  ☆34Updated last month
• nickjiang2378 / test-time-registers
  [NeurIPS '25 Spotlight] Official Pytorch implementation of "Vision Transformers Don't Need Trained Registers"
  ☆149Updated 2 months ago
• ShadeAlsha / ICon
  ICLR 2025 - official implementation for "I-Con: A Unifying Framework for Representation Learning"
  ☆117Updated 5 months ago
• hyperevolnet / Terminator
  The official repository for HyperZ⋅Z⋅W Operator Connects Slow-Fast Networks for Full Context Interaction.
  ☆42Updated 8 months ago
• AhmedImtiazPrio / grok-adversarial
  Deep Networks Grok All the Time and Here is Why
  ☆38Updated last year
• lucidrains / spline-based-transformer
  Implementation of the proposed Spline-Based Transformer from Disney Research
  ☆105Updated last year
• james-oldfield / muMoE
  [NeurIPS'24] Multilinear Mixture of Experts: Scalable Expert Specialization through Factorization
  ☆37Updated last year
• TomerRonen34 / mixed-resolution-vit
  ☆56Updated 2 years ago
• giannisdaras / ambient-tweedie
  [ICML 2024]: Official implementation for the paper: "Consistent Diffusion Meets Tweedie"
  ☆53Updated last year
• ml-jku / MIM-Refiner
  A Contrastive Learning Boost from Intermediate Pre-Trained Representations
  ☆43Updated last year
• ethanbar11 / ssm_2d
  More dimensions = More fun
  ☆26Updated last year
• SriramB-98 / vit-decompose
  ☆23Updated 11 months ago
• amirbar / StoP
  ☆12Updated last year
• goombalab / hydra
  Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"
  ☆166Updated 10 months ago
• oripress / EntropyEnigma
  Official code for the ICML 2024 paper "The Entropy Enigma: Success and Failure of Entropy Minimization"
  ☆55Updated last year
• samar-khanna / ExPLoRA
  Official code repository for ICML 2025 paper: "ExPLoRA: Parameter-Efficient Extended Pre-training to Adapt Vision Transformers under Doma…
  ☆47Updated 3 months ago
• lucidrains / gradnorm-pytorch
  A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch
  ☆120Updated 3 months ago
• davidpicard / HoMM
  High order Moment Models
  ☆42Updated 3 weeks ago
• CompVis / DisCLIP
  [AAAI 2025] Does VLM Classification Benefit from LLM Description Semantics?
  ☆24Updated 4 months ago
• alhojel / visual_task_vectors
  ☆40Updated last year
• google-deepmind / tips
  ☆107Updated 8 months ago
• Westlake-AI / SEMA
  Switch EMA: A Free Lunch for Better Flatness and Sharpness
  ☆28Updated last year
• Adamdad / neumeta
  NeuMeta transforms neural networks by allowing a single model to adapt on the fly to different sizes, generating the right weights when n…
  ☆43Updated last year
• EPFL-VILAB / fm-vision-evals
  ☆72Updated 4 months ago
• OpenNLPLab / HGRN2
  HGRN2: Gated Linear RNNs with State Expansion
  ☆55Updated last year