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 11 months ago
• knsiczarnamagia / wave
  ☆27Updated 10 months ago
• clu0 / unet.cu
  UNet diffusion model in pure CUDA
  ☆651Updated last year
• HomebrewML / HeavyBall
  Efficient optimizers
  ☆276Updated 3 weeks ago
• cloneofsimo / minRF
  Minimal implementation of scalable rectified flow transformers, based on SD3's approach
  ☆613Updated last year
• nanowell / AdEMAMix-Optimizer-Pytorch
  The AdEMAMix Optimizer: Better, Faster, Older.
  ☆186Updated last year
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆293Updated last year
• ironjr / grokfast
  Official repository for the paper "Grokfast: Accelerated Grokking by Amplifying Slow Gradients"
  ☆564Updated last year
• ethansmith2000 / fsdp_optimizers
  supporting pytorch FSDP for optimizers
  ☆83Updated 11 months ago
• lucidrains / nGPT-pytorch
  Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI
  ☆291Updated 5 months ago
• zyushun / Adam-mini
  Code for Adam-mini: Use Fewer Learning Rates To Gain More https://arxiv.org/abs/2406.16793
  ☆440Updated 5 months ago
• lucidrains / ring-attention-pytorch
  Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch
  ☆542Updated 5 months ago
• Algomancer / Bayesian-Flow-Networks
  A simple implimentation of Bayesian Flow Networks (BFN)
  ☆240Updated last year
• ash-01xor / bpe.c
  Simple Byte pair Encoding mechanism used for tokenization process . written purely in C
  ☆139Updated 11 months ago
• BobMcDear / attorch
  A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.
  ☆580Updated 2 months ago
• cloneofsimo / vqgan-training
  Train VAE like a boss
  ☆297Updated last year
• dingo-actual / infini-transformer
  PyTorch implementation of Infini-Transformer from "Leave No Context Behind: Efficient Infinite Context Transformers with Infini-attention…
  ☆292Updated last year
• wolfecameron / nanoMoE
  An extension of the nanoGPT repository for training small MOE models.
  ☆207Updated 8 months ago
• hkproj / triton-flash-attention
  ☆215Updated 10 months ago
• KindXiaoming / grow-crystals
  Getting crystal-like representations with harmonic loss
  ☆192Updated 7 months ago
• explainingai-code / DiT-PyTorch
  This repo implements Diffusion Transformers(DiT) in PyTorch and provides training and inference code on CelebHQ dataset
  ☆50Updated 10 months ago
• srush / annotated-mamba
  Annotated version of the Mamba paper
  ☆490Updated last year
• isamu-isozaki / huggingface-reading-group
  This repository's goal is to precompile all past presentations of the Huggingface reading group
  ☆48Updated last year
• HazyResearch / m2
  Repo for "Monarch Mixer: A Simple Sub-Quadratic GEMM-Based Architecture"
  ☆560Updated 10 months ago
• apapiu / transformer_latent_diffusion
  Text to Image Latent Diffusion using a Transformer core
  ☆212Updated last year
• KellerJordan / cifar10-airbench
  CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds
  ☆321Updated 3 months ago
• microsoft / Samba
  [ICLR 2025] Samba: Simple Hybrid State Space Models for Efficient Unlimited Context Language Modeling
  ☆923Updated last week
• danielvegamyhre / ml-perf-reading-group
  EleutherAI ML Performance reading group repository (slides, meeting recordings, annotated papers)
  ☆21Updated last week
• KellerJordan / Muon
  Muon is an optimizer for hidden layers in neural networks
  ☆1,960Updated 3 months ago
• NVIDIA / ngpt
  Normalized Transformer (nGPT)
  ☆192Updated 11 months ago