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!
  ☆14Updated 4 months ago
• knsiczarnamagia / wave3
  ☆27Updated 3 months ago
• HomebrewML / HeavyBall
  Efficient optimizers
  ☆186Updated this week
• apapiu / transformer_latent_diffusion
  Text to Image Latent Diffusion using a Transformer core
  ☆175Updated 7 months ago
• cloneofsimo / minRF
  Minimal implementation of scalable rectified flow transformers, based on SD3's approach
  ☆502Updated 9 months ago
• ironjr / grokfast
  Official repository for the paper "Grokfast: Accelerated Grokking by Amplifying Slow Gradients"
  ☆553Updated 9 months ago
• cloneofsimo / vqgan-training
  Train VAE like a boss
  ☆274Updated 5 months ago
• zyushun / Adam-mini
  Code for Adam-mini: Use Fewer Learning Rates To Gain More https://arxiv.org/abs/2406.16793
  ☆400Updated 4 months ago
• Algomancer / Bayesian-Flow-Networks
  A simple implimentation of Bayesian Flow Networks (BFN)
  ☆240Updated last year
• KellerJordan / cifar10-airbench
  CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds
  ☆226Updated last month
• cabralpinto / modular-diffusion
  Python library for designing and training your own Diffusion Models with PyTorch.
  ☆279Updated 8 months ago
• kuleshov-group / awesome-discrete-diffusion-models
  A curated list for awesome discrete diffusion models resources.
  ☆285Updated 2 months ago
• tanelp / tiny-diffusion
  A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.
  ☆727Updated 11 months ago
• kuleshov-group / mdlm
  Simple and Effective Masked Diffusion Language Model
  ☆362Updated last month
• louaaron / Score-Entropy-Discrete-Diffusion
  [ICML 2024 Best Paper] Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution (https://arxiv.org/abs/2310.16834)
  ☆543Updated last year
• jzhang38 / LongMamba
  Some preliminary explorations of Mamba's context scaling.
  ☆212Updated last year
• madaan / minimal-text-diffusion
  A minimal implementation of diffusion models for text generation
  ☆356Updated last year
• ethansmith2000 / fsdp_optimizers
  supporting pytorch FSDP for optimizers
  ☆80Updated 4 months ago
• lucidrains / rectified-flow-pytorch
  Implementation of rectified flow and some of its followup research / improvements in Pytorch
  ☆272Updated 2 months ago
• NVlabs / edm2
  EDM2 and Autoguidance -- Official PyTorch implementation
  ☆676Updated 4 months ago
• KindXiaoming / grow-crystals
  Getting crystal-like representations with harmonic loss
  ☆180Updated last week
• albarji / toy-diffusion
  A toy implementation of a diffusion model for low-dimensional data
  ☆23Updated last year
• srush / annotated-mamba
  Annotated version of the Mamba paper
  ☆480Updated last year
• AssemblyAI-Community / MinImagen
  MinImagen: A minimal implementation of the Imagen text-to-image model
  ☆302Updated last year
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆270Updated 10 months ago
• pytorch-labs / attention-gym
  Helpful tools and examples for working with flex-attention
  ☆707Updated this week
• SwayStar123 / microdiffusion
  ☆44Updated last month
• lucidrains / recurrent-interface-network-pytorch
  Implementation of Recurrent Interface Network (RIN), for highly efficient generation of images and video without cascading networks, in P…
  ☆202Updated last year
• junhsss / consistency-models
  A Toolkit for OpenAI's Consistency Models.
  ☆204Updated last year
• lebellig / discrete-fm
  Educational implementation of the Discrete Flow Matching paper
  ☆82Updated 7 months ago