lucidrains / titans-pytorchLinks
Unofficial implementation of Titans, SOTA memory for transformers, in Pytorch
☆1,748Updated this week
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☆2,116Updated last month
- Code for BLT research paper☆2,018Updated last month
- A Self-adaptation Framework🐙 that adapts LLMs for unseen tasks in real-time!☆1,177Updated 10 months ago
- Official PyTorch implementation of Learning to (Learn at Test Time): RNNs with Expressive Hidden States☆1,294Updated last year
- Official PyTorch implementation for "Large Language Diffusion Models"☆3,424Updated last month
- H-Net: Hierarchical Network with Dynamic Chunking☆797Updated last month
- [ICLR2025 Spotlight🔥] Official Implementation of TokenFormer: Rethinking Transformer Scaling with Tokenized Model Parameters☆579Updated 10 months ago
- Pretraining and inference code for a large-scale depth-recurrent language model☆856Updated 2 months ago
- Implementation of the sparse attention pattern proposed by the Deepseek team in their "Native Sparse Attention" paper☆790Updated 4 months ago
- [ICLR 2025] Samba: Simple Hybrid State Space Models for Efficient Unlimited Context Language Modeling☆933Updated last month
- Continuous Thought Machines, because thought takes time and reasoning is a process.☆1,640Updated last week
- 🚀 Efficient implementations of state-of-the-art linear attention models☆4,089Updated this week
- Implementing DeepSeek R1's GRPO algorithm from scratch☆1,718Updated 8 months ago
- Training Large Language Model to Reason in a Continuous Latent Space☆1,411Updated 4 months ago
- Dream 7B, a large diffusion language model☆1,115Updated last month
- Code release for DynamicTanh (DyT)☆1,028Updated 8 months ago
- A simple and efficient Mamba implementation in pure PyTorch and MLX.☆1,387Updated last year
- MMaDA - Open-Sourced Multimodal Large Diffusion Language Models☆1,532Updated last month
- dLLM: Simple Diffusion Language Modeling☆1,504Updated this week
- [ICLR 2025 Oral] Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Models☆920Updated 5 months ago
- Recipes to scale inference-time compute of open models☆1,120Updated 7 months ago
- Official JAX implementation of Learning to (Learn at Test Time): RNNs with Expressive Hidden States☆438Updated last month
- Muon is Scalable for LLM Training☆1,387Updated 4 months ago
- Mixture-of-Recursions: Learning Dynamic Recursive Depths for Adaptive Token-Level Computation (NeurIPS 2025)☆526Updated 3 months ago
- GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projection☆1,637Updated last year
- SONAR, a new multilingual and multimodal fixed-size sentence embedding space, with a full suite of speech and text encoders and decoders.☆851Updated 2 months ago
- [NeurIPS 2025 Spotlight] TPA: Tensor ProducT ATTenTion Transformer (T6) (https://arxiv.org/abs/2501.06425)☆435Updated last week
- [ICML2024 (Oral)] Official PyTorch implementation of DoRA: Weight-Decomposed Low-Rank Adaptation☆897Updated last year
- PyTorch code and models for VJEPA2 self-supervised learning from video.☆2,586Updated 3 months ago
- PyTorch Code for Energy-Based Transformers paper -- generalizable reasoning and scalable learning☆568Updated last month