lucidrains repositories

Implementation of the general framework for AMIE, from the paper "Towards Conversational Diagnostic AI", out of Google Deepmind

Last commit Sep 16, 2024

Prediction of key transcription factors in cell fate determination using enhancer networks. See full ANANSE documentation for detailed installation instructions and usage examples. http://anansepy.readthedocs.io

Last commit Sep 28, 2021

Implementation of Attention Schema-based Attention Control (ASAC), A Cognitive-Inspired Approach for Attention Management in Transformers

Last commit Jun 9, 2026

Adaptive and Momental Bounds for Adaptive Learning Rate Methods.

Last commit Nov 13, 2019

Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch

Last commit Sep 1, 2022

A Pytorch implementation of Attention on Attention module (both self and guided variants), for Visual Question Answering

Last commit Nov 8, 2020

Implementation of Band Split Roformer, SOTA Attention network for music source separation out of ByteDance AI Labs

Last commit Feb 1, 2026

Codebase for Berkeley Humanoid Lite

Last commit Sep 7, 2025

Implementation of CALM from the paper "LLM Augmented LLMs: Expanding Capabilities through Composition", out of Google Deepmind

Last commit Sep 12, 2024

Contrastive Language-Audio Pretraining

Last commit May 20, 2021

Contrastive Language-Image Pretraining

Last commit Mar 3, 2023

Implementation of CoCa, Contrastive Captioners are Image-Text Foundation Models, in Pytorch

Last commit Dec 12, 2023

Implementation of the conditionally routed attention in the CoLT5 architecture, in Pytorch

Last commit Sep 6, 2024

Making Protein folding accessible to all via Google Colab!

Last commit Mar 18, 2022

Open-AI's DALL-E for large scale training in mesh-tensorflow.

Last commit Jan 17, 2021

Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch

Last commit Sep 16, 2023

Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch

Last commit Oct 19, 2023

Implementation of DASH, Warm-Starting Neural Network Training in Stationary Settings without Loss of Plasticity

Last commit Oct 17, 2025

Training and testing of DeepBind models.

Last commit Oct 19, 2015

Implementation of Dex1B: Learning with 1B Demonstrations for Dexterous Manipulation, from Ye et al of UCSD

Last commit Aug 9, 2025