This repo implements Diffusion Transformers(DiT) in PyTorch and provides training and inference code on CelebHQ dataset
☆69Jan 6, 2025Updated last year
Alternatives and similar repositories for DiT-PyTorch
Users that are interested in DiT-PyTorch are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- This repo implements Video generation model using Latent Diffusion Transformers(Latte) in PyTorch and provides training and inference cod…☆19Jan 6, 2025Updated last year
- This repo implements Denoising Diffusion Probabilistic Models (DDPM) in Pytorch☆187Nov 25, 2024Updated last year
- Pytorch Implementation of the paper: "Learning to (Learn at Test Time): RNNs with Expressive Hidden States"☆25Jun 22, 2026Updated last week
- This repo implements a Stable Diffusion model in PyTorch with all the essential components.☆254Nov 24, 2024Updated last year
- stream-of-consciousness experience of an AI's thinking process, complete with creative tangents and unexpected connections.☆14Jan 29, 2025Updated last year
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- ☆12Dec 23, 2024Updated last year
- Cryptanalysis tool for the Enigma machine☆17Aug 21, 2023Updated 2 years ago
- official implementation of "CLIP-VQDiffusion : Langauge Free Training of Text To Image generation using CLIP and vector quantized diffusi…☆19Sep 5, 2024Updated last year
- A curated collection of prompts for Grok Imagine by xAI☆32Jun 6, 2026Updated 3 weeks ago
- ☆60Mar 22, 2024Updated 2 years ago
- ☆46May 24, 2025Updated last year
- Mixture of Experts from scratch☆14Apr 12, 2024Updated 2 years ago
- This repository implements a simpleVAE for training on CPU on the MNIST dataset and provides ability to visualize the latent space, entir…☆43Jul 11, 2024Updated last year
- See https://github.com/cuda-mode/triton-index/ instead!☆11May 8, 2024Updated 2 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- The open source implementation of the multi grouped query attention by the paper "GQA: Training Generalized Multi-Query Transformer Model…☆16Dec 11, 2023Updated 2 years ago
- [BMM 24-25] HippoTrainer: Gradient-Based Hyperparameter Optimization☆11May 6, 2025Updated last year
- D-JEPA on ImageNet☆23Nov 18, 2024Updated last year
- Unsupervised Lifelong Person Re-identification via Contrastive Rehearsal☆11Apr 7, 2022Updated 4 years ago
- code for paper “PET Image Denoising with Score-Based Diffusion Probabilistic Models”☆11Dec 23, 2023Updated 2 years ago
- Rembg is a tool to remove images background.☆12Nov 29, 2022Updated 3 years ago
- ☆15Jul 24, 2024Updated last year
- Cutting-edge Python library designed for generative image augmentation!☆13Mar 25, 2025Updated last year
- ☆16Aug 7, 2024Updated last year
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- ☆19Jul 31, 2025Updated 10 months ago
- ☆36Jan 10, 2026Updated 5 months ago
- [NeurIPS 2025 Spotlight] Implementation of "KLASS: KL-Guided Fast Inference in Masked Diffusion Models"☆32Jan 3, 2026Updated 5 months ago
- ☆12Dec 14, 2024Updated last year
- Implementation of Dat2Vec2.0 for vision☆18Feb 6, 2023Updated 3 years ago
- ☆26Oct 15, 2024Updated last year
- Wan: Open and Advanced Large-Scale Video Generative Models☆31Jul 28, 2025Updated 11 months ago
- SpeechPlus: Small LLM-Based Text-to-Speech Library 🚀☆21May 20, 2025Updated last year
- This is a pytorch implementation of Denoising Diffusion Implicit Models☆136Jul 10, 2023Updated 2 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- [ICCV 2025 Highlight] VolumetricSMPL☆94Jul 27, 2025Updated 11 months ago
- Python client for Jikan.moe, MyAnimeList unofficial API with good intentions.☆14Dec 20, 2022Updated 3 years ago
- 🎈 一个简洁大气，含夜间模式的 halo 博客主题☆20Jun 2, 2024Updated 2 years ago
- BMM☆13May 21, 2026Updated last month
- ☆16Jul 8, 2024Updated last year
- A volumetric version of the HCP-MMP1 atlas conveniently imported in FSL. Please consider citing "Bedini, M., Olivetti, E., Avesani, P., &…☆29Mar 5, 2025Updated last year
- Training framework for Large Behavioral Models☆28Sep 17, 2025Updated 9 months ago