lucidrains repositories

A Tiny, Pure Javascript implementation of Gradient Boosted Trees.

No recent commit data

Unofficial implementation of Titans, SOTA memory for transformers, in Pytorch

Last commit Jun 6, 2026

Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"

Last commit Jun 20, 2024

Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing

Last commit Jan 27, 2022

Implementation of Toolformer, Language Models That Can Use Tools, by MetaAI

Last commit Dec 21, 2023

Some utility functions to help myself (and perhaps others) go faster with ML/AI work

Last commit Jun 2, 2026

Implementation of trRosetta and trDesign for Pytorch, made into a convenient package, for protein structure prediction and design

Last commit Aug 27, 2021

Implementation of Tranception, an attention network, paired with retrieval, that is SOTA for protein fitness prediction

Last commit Jun 19, 2022

Explorations into whether a transformer with RL can direct a genetic algorithm to converge faster

Last commit Jun 2, 2026

Implementation of Transformer in Transformer, pixel level attention paired with patch level attention for image classification, in Pytorch

Last commit Dec 27, 2021

Explorations into adversarial losses on top of autoregressive loss for language modeling

Last commit Dec 21, 2025

Implementation of Transframer, Deepmind's U-net + Transformer architecture for up to 30 seconds video generation, in Pytorch

Last commit Aug 23, 2022

Pytorch implementation of Transfusion, "Predict the Next Token and Diffuse Images with One Multi-Modal Model", from MetaAI

Last commit Jan 27, 2026

Implementation of TransGanFormer, an all-attention GAN that combines the finding from the recent GanFormer and TransGan paper

Last commit Apr 27, 2021

Implementation of the Triangle Multiplicative module, used in Alphafold2 as an efficient way to mix rows or columns of a 2d feature map, as a standalone package for Pytorch

Last commit Aug 3, 2021

Development repository for the Triton language and compiler

Last commit Aug 10, 2022

Implementation of a Transformer, but completely in Triton

Last commit Apr 5, 2022

Reinforcement Learning example in Nim, playing tic tac toe. Based off original C version from the great Antirez

Last commit Apr 2, 2025

Applying NLP transfer learning techniques to predict Tweet stance

Last commit Feb 10, 2019

Explorations into the proposal from the paper, Visual-tactile pretraining and online multitask learning for humanlike manipulation dexterity

Last commit Feb 6, 2026