Alternatives and similar repositories for PSI

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

• knsiczarnamagia / kaggle-tutorial
  Kick-off repository for starting with Kaggle!
  ☆12Updated last year
• knsiczarnamagia / wave
  ☆27Updated 11 months ago
• HomebrewML / HeavyBall
  Efficient optimizers
  ☆277Updated last month
• cloneofsimo / minRF
  Minimal implementation of scalable rectified flow transformers, based on SD3's approach
  ☆624Updated last year
• zyushun / Adam-mini
  Code for Adam-mini: Use Fewer Learning Rates To Gain More https://arxiv.org/abs/2406.16793
  ☆445Updated 7 months ago
• hkproj / triton-flash-attention
  ☆227Updated 11 months ago
• clu0 / unet.cu
  UNet diffusion model in pure CUDA
  ☆656Updated last year
• danielvegamyhre / ml-perf-reading-group
  EleutherAI ML Performance reading group repository (slides, meeting recordings, annotated papers)
  ☆23Updated this week
• ethansmith2000 / fsdp_optimizers
  supporting pytorch FSDP for optimizers
  ☆84Updated last year
• Cornell-RelaxML / quip-sharp
  ☆574Updated last year
• sayakpaul / diffusers-torchao
  End-to-end recipes for optimizing diffusion models with torchao and diffusers (inference and FP8 training).
  ☆388Updated 6 months ago
• ironjr / grokfast
  Official repository for the paper "Grokfast: Accelerated Grokking by Amplifying Slow Gradients"
  ☆565Updated last year
• Nota-NetsPresso / BK-SDM
  A Compressed Stable Diffusion for Efficient Text-to-Image Generation [ECCV'24]
  ☆305Updated last year
• cloneofsimo / vqgan-training
  Train VAE like a boss
  ☆307Updated last year
• apapiu / transformer_latent_diffusion
  Text to Image Latent Diffusion using a Transformer core
  ☆216Updated last year
• srush / annotated-mamba
  Annotated version of the Mamba paper
  ☆492Updated last year
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆298Updated last year
• KellerJordan / cifar10-airbench
  CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds
  ☆335Updated last month
• a-hamdi / native-sparse-attention
  ☆15Updated 9 months ago
• rkinas / triton-resources
  A curated list of resources for learning and exploring Triton, OpenAI's programming language for writing efficient GPU code.
  ☆441Updated 9 months ago
• microsoft / dion
  Dion optimizer algorithm
  ☆404Updated this week
• PeaBrane / mamba-tiny
  Simple, minimal implementation of the Mamba SSM in one pytorch file. Using logcumsumexp (Heisen sequence).
  ☆128Updated last year
• spcl / QuaRot
  Code for Neurips24 paper: QuaRot, an end-to-end 4-bit inference of large language models.
  ☆466Updated last year
• LambdaLabsML / distributed-training-guide
  Best practices & guides on how to write distributed pytorch training code
  ☆552Updated last month
• BobMcDear / attorch
  A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.
  ☆585Updated 4 months ago
• hkproj / multi-latent-attention
  ☆45Updated 6 months ago
• explainingai-code / DiT-PyTorch
  This repo implements Diffusion Transformers(DiT) in PyTorch and provides training and inference code on CelebHQ dataset
  ☆53Updated 11 months ago
• tmabraham / diffusion_reading_group
  Diffusion Reading Group at EleutherAI
  ☆333Updated 2 years ago
• meta-pytorch / attention-gym
  Helpful tools and examples for working with flex-attention
  ☆1,089Updated this week
• Cornell-RelaxML / QuIP
  Code for paper: "QuIP: 2-Bit Quantization of Large Language Models With Guarantees"
  ☆391Updated last year