Alternatives and similar repositories for pytorch

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

• KONAKONA666 / q8_kernels
  ☆79Updated last year
• huggingface / flux-fast
  Making Flux go brrr on GPUs.
  ☆161Updated last month
• cloneofsimo / min-max-in-dit
  ☆27Updated last year
• yangluo7 / CAME
  [ACL 2023] The official implementation of "CAME: Confidence-guided Adaptive Memory Optimization"
  ☆96Updated 10 months ago
• timudk / flux_triton
  Writing FLUX in Triton
  ☆41Updated last year
• cloneofsimo / imagenet.int8
  ☆39Updated last year
• mlfoundations / open-diffusion
  Simple large-scale training of stable diffusion with multi-node support.
  ☆133Updated 2 years ago
• ai-compiler-study / triton-kernels
  Triton kernels for Flux
  ☆22Updated 7 months ago
• meta-pytorch / superblock
  A block oriented training approach for inference time optimization.
  ☆34Updated last year
• Birch-san / diffusers-play
  Repository with which to explore k-diffusion and diffusers, and within which changes to said packages may be tested.
  ☆55Updated 2 years ago
• fal-ai / diffusion-speedrun
  Focused on fast experimentation and simplicity
  ☆80Updated last year
• AndyShih12 / paradigms
  PyTorch implementation for "Parallel Sampling of Diffusion Models", NeurIPS 2023 Spotlight
  ☆153Updated 2 years ago
• mit-han-lab / patch_conv
  Patch convolution to avoid large GPU memory usage of Conv2D
  ☆95Updated last year
• sandyresearch / chipmunk
  🎬 3.7× faster video generation E2E 🖼️ 1.6× faster image generation E2E ⚡ ColumnSparseAttn 9.3× vs FlashAttn‑3 💨 ColumnSparseGEMM 2.5× …
  ☆101Updated 5 months ago
• aredden / torch-cublas-hgemm
  PyTorch half precision gemm lib w/ fused optional bias + optional relu/gelu
  ☆78Updated last year
• fal-ai-community / NativeSparseAttention
  research impl of Native Sparse Attention (2502.11089)
  ☆63Updated 11 months ago
• apple / ml-vision-transformers-ane
  ☆91Updated 2 years ago
• yandex-research / vqdm
  Official repository for VQDM:Accurate Compression of Text-to-Image Diffusion Models via Vector Quantization paper
  ☆34Updated last year
• HazyResearch / diffusers
  🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch
  ☆50Updated 3 years ago
• Lucky-Lance / TerDiT
  TerDiT: Ternary Diffusion Models with Transformers
  ☆74Updated last year
• NVlabs / T-Stitch
  [ICLR 2025] Official PyTorch implmentation of paper "T-Stitch: Accelerating Sampling in Pre-trained Diffusion Models with Trajectory Stit…
  ☆104Updated last year
• OFA-Sys / diffusion-deploy
  ☆52Updated 2 years ago
• facebookresearch / SSDD
  Official implementation for SSDD Single-Step Diffusion Decoder for Efficient Image Tokenization.
  ☆53Updated 2 months ago
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆29Updated last year
• sayakpaul / nanoDiT
  Just another reasonably minimal repo for class-conditional training of pixel-space diffusion transformers.
  ☆143Updated 8 months ago
• cloneofsimo / ezmup
  Simple implementation of muP, based on Spectral Condition for Feature Learning. The implementation is SGD only, dont use it for Adam
  ☆85Updated last year
• ziplab / PTQD
  The official implementation of PTQD: Accurate Post-Training Quantization for Diffusion Models
  ☆103Updated last year
• vdesai2014 / inference-optimization-blog-post
  ☆91Updated last year
• eai-lab / On-device-Sora
  [arXiv] On-device Sora: Enabling Diffusion-Based Text-to-Video Generation for Mobile Devices
  ☆131Updated 2 months ago
• cloneofsimo / karras-power-ema-tutorial
  ☆53Updated 2 years ago