lucidrains/light-recurrent-unit-pytorch

LSTM, RNN and GRU implementations using Pytorch

Implementation of the recurrent looped transformer proposed by Yifan Zhang of Princeton

FlashRNN - Fast RNN Kernels with I/O Awareness

Official repository of the xLSTM.

PyTorch and TensorFlow implementation of NCP, LTC, and CfC wired neural models

An implementation of "Retentive Network: A Successor to Transformer for Large Language Models"

A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training

Implementation of rectified flow and some of its followup research / improvements in Pytorch

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.