Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper
Repositories
lucidrains repositories
Implementation and explorations into PopuLoRA, Co-Evolving LLM Populations for Reasoning Self-Play
An implementation of PPO in Pytorch
Standalone Product Key Memory module in Pytorch - for augmenting Transformer models
Implementation and replication of ProGen, Language Modeling for Protein Generation, in Jax
Implementation of ProteinBERT in Pytorch
Implementation of the pseudo projector proposed by Vitaly Bulgakov at Mass General Brigham
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new AI research
torch-optimizer -- collection of optimizers for Pytorch
Implementation of Q-Transformer, Scalable Offline Reinforcement Learning via Autoregressive Q-Functions, out of Google Deepmind
Exploring an idea where one forgets about efficiency and carries out attention across each edge of the nodes (tokens)
Implementation of rectified flow and some of its followup research / improvements in Pytorch
Implementation of Recurrent Interface Network (RIN), for highly efficient generation of images and video without cascading networks, in Pytorch
Implementation of Recurrent Memory Transformer, Neurips 2022 paper, in Pytorch
Deep Recurrent Neural Networks and LSTMs in Javascript. More generally also arbitrary expression graphs with automatic differentiation.
Reformer, the efficient Transformer, in Pytorch
Implementation of a Transformer using ReLA (Rectified Linear Attention) from https://arxiv.org/abs/2104.07012
Implementation of the Remixer Block from the Remixer paper, in Pytorch
Implementation of ResMLP, an all MLP solution to image classification, in Pytorch