Westlake-AI / SEMA
Switch EMA: A Free Lunch for Better Flatness and Sharpness
☆26Updated last year
Alternatives and similar repositories for SEMA:
Users that are interested in SEMA are comparing it to the libraries listed below
- This is a PyTorch implementation of the paperViP A Differentially Private Foundation Model for Computer Vision☆36Updated last year
- [ECCV 2024] Official pytorch implementation of "Switch Diffusion Transformer: Synergizing Denoising Tasks with Sparse Mixture-of-Experts"☆38Updated 8 months ago
- A big_vision inspired repo that implements a generic Auto-Encoder class capable in representation learning and generative modeling.☆34Updated 9 months ago
- ☆50Updated last year
- This is a offical PyTorch/GPU implementation of SupMAE.☆77Updated 2 years ago
- An official PyTorch implementation for CLIPPR☆29Updated last year
- ☆23Updated 5 months ago
- [ECCV 2024] This is the official implementation of "Stitched ViTs are Flexible Vision Backbones".☆27Updated last year
- PyTorch Implementation of "Your ViT is Secretly a Hybrid Discriminative-Generative Diffusion Model"☆48Updated 2 years ago
- ☆30Updated 2 months ago
- Code for the paper Self-Supervised Learning of Split Invariant Equivariant Representations☆28Updated last year
- Minimal multi-gpu implementation of EDM2: "Analyzing and Improving the Training Dynamics of Diffusion Models"☆29Updated last year
- Locally Hierarchical Auto-Regressive Modeling for Image Generation (HQ-Transformer)☆29Updated last year
- Code release for paper Extremely Simple Activation Shaping for Out-of-Distribution Detection☆52Updated 7 months ago
- Repo for the paper "Extrapolating from a Single Image to a Thousand Classes using Distillation"☆36Updated 8 months ago
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆106Updated 5 months ago
- [CVPR 2024 Highlight] OpenBias: Open-set Bias Detection in Text-to-Image Generative Models☆22Updated last month
- [NeurIPS 2024] VeLoRA : Memory Efficient Training using Rank-1 Sub-Token Projections☆19Updated 5 months ago
- Code base of SynthCLIP: CLIP training with purely synthetic text-image pairs from LLMs and TTIs.☆97Updated last week
- ☆37Updated 8 months ago
- Code for T-MARS data filtering☆35Updated last year
- [NeurIPS'24] Multilinear Mixture of Experts: Scalable Expert Specialization through Factorization☆29Updated 6 months ago
- Augmenting with Language-guided Image Augmentation (ALIA)☆75Updated last year
- [CVPR 2025] Official PyTorch implementation of MaskSub "Masking meets Supervision: A Strong Learning Alliance"☆35Updated this week
- [Preprint] GMem: A Modular Approach for Ultra-Efficient Generative Models☆29Updated 2 weeks ago
- ☆40Updated 4 months ago
- ☆42Updated 2 weeks ago
- ☆23Updated 5 months ago
- The official repository of paper "ScaleLong: Towards More Stable Training of Diffusion Model via Scaling Network Long Skip Connection" (N…☆50Updated last year
- source code for NeurIPS'23 paper "Dream the Impossible: Outlier Imagination with Diffusion Models"☆67Updated 2 months ago