lucidrains / titans-pytorchLinks
Unofficial implementation of Titans, SOTA memory for transformers, in Pytorch
☆1,486Updated last month
Alternatives and similar repositories for titans-pytorch
Users that are interested in titans-pytorch are comparing it to the libraries listed below
Sorting:
- Muon is an optimizer for hidden layers in neural networks☆1,983Updated 4 months ago
- Official PyTorch implementation of Learning to (Learn at Test Time): RNNs with Expressive Hidden States☆1,273Updated last year
- Code for BLT research paper☆2,006Updated last week
- Continuous Thought Machines, because thought takes time and reasoning is a process.☆1,396Updated 3 weeks ago
- Official PyTorch implementation for "Large Language Diffusion Models"☆3,210Updated this week
- 🚀 Efficient implementations of state-of-the-art linear attention models☆3,787Updated last week
- A Self-adaptation Framework🐙 that adapts LLMs for unseen tasks in real-time!☆1,162Updated 9 months ago
- Implementation of the sparse attention pattern proposed by the Deepseek team in their "Native Sparse Attention" paper☆779Updated 2 months ago
- Pretraining and inference code for a large-scale depth-recurrent language model☆843Updated 3 weeks ago
- [ICLR2025 Spotlight🔥] Official Implementation of TokenFormer: Rethinking Transformer Scaling with Tokenized Model Parameters☆576Updated 9 months ago
- [ICLR 2025] Samba: Simple Hybrid State Space Models for Efficient Unlimited Context Language Modeling☆926Updated last week
- H-Net: Hierarchical Network with Dynamic Chunking☆769Updated last month
- Muon is Scalable for LLM Training☆1,354Updated 3 months ago
- Implementing DeepSeek R1's GRPO algorithm from scratch☆1,662Updated 6 months ago
- Training Large Language Model to Reason in a Continuous Latent Space☆1,327Updated 3 months ago
- Mixture-of-Recursions: Learning Dynamic Recursive Depths for Adaptive Token-Level Computation (NeurIPS 2025)☆504Updated last month
- Code release for DynamicTanh (DyT)☆1,020Updated 7 months ago
- A simple and efficient Mamba implementation in pure PyTorch and MLX.☆1,360Updated 11 months ago
- Dream 7B, a large diffusion language model☆1,054Updated last month
- Build high-performance AI models with modular building blocks☆562Updated this week
- Official JAX implementation of Learning to (Learn at Test Time): RNNs with Expressive Hidden States☆424Updated last week
- NanoGPT (124M) in 3 minutes☆3,785Updated last week
- [ICLR 2025 Oral] Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Models☆884Updated 4 months ago
- [ICML2024 (Oral)] Official PyTorch implementation of DoRA: Weight-Decomposed Low-Rank Adaptation☆875Updated last year
- [NeurIPS 2025] MMaDA - Open-Sourced Multimodal Large Diffusion Language Models☆1,473Updated last month
- PyTorch code and models for VJEPA2 self-supervised learning from video.☆2,399Updated 2 months ago
- [NeurIPS 2025 Spotlight] TPA: Tensor ProducT ATTenTion Transformer (T6) (https://arxiv.org/abs/2501.06425)☆422Updated 2 weeks ago
- MoBA: Mixture of Block Attention for Long-Context LLMs☆1,969Updated 7 months ago
- Large Concept Models: Language modeling in a sentence representation space☆2,302Updated 9 months ago
- The simplest, fastest repository for training/finetuning small-sized VLMs.☆4,229Updated 2 weeks ago