Implementation of MEGABYTE, Predicting Million-byte Sequences with Multiscale Transformers, in Pytorch
☆658Dec 27, 2024Updated last year
Alternatives and similar repositories for MEGABYTE-pytorch
Users that are interested in MEGABYTE-pytorch are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Experiments around a simple idea for inducing multiple hierarchical predictive model within a GPT☆230Sep 23, 2026Updated last month
- Implementation of Recurrent Memory Transformer, Neurips 2022 paper, in Pytorch☆425Jan 6, 2025Updated last year
- My own attempt at a long context genomics model, leveraging recent advances in long context attention modeling (Flash Attention + other h…☆54Jul 2, 2023Updated 3 years ago
- Implementation of the conditionally routed attention in the CoLT5 architecture, in Pytorch☆231Aug 14, 2026Updated 2 months ago
- RWKV (pronounced RwaKuv) is an RNN with great LLM performance, which can also be directly trained like a GPT transformer (parallelizable)…☆14,767Updated this week
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Convolutions for Sequence Modeling☆922Jun 13, 2024Updated 2 years ago
- Implementation of GateLoop Transformer in Pytorch and Jax☆93Jun 18, 2024Updated 2 years ago
- My attempts at applying Soundstream design on learned tokenization of text and then applying hierarchical attention to text generation☆90Oct 11, 2024Updated 2 years ago
- Public repo for the NeurIPS 2023 paper "Unlimiformer: Long-Range Transformers with Unlimited Length Input"☆1,061Mar 7, 2024Updated 2 years ago
- The official implementation of “Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-training”☆1,000Jan 30, 2024Updated 2 years ago
- Implementation of MambaByte in "MambaByte: Token-free Selective State Space Model" in Pytorch and Zeta☆129Updated this week
- A concise but complete full-attention transformer with a set of promising experimental features from various papers☆5,961Oct 23, 2026Updated last week
- Some personal experiments around routing tokens to different autoregressive attention, akin to mixture-of-experts☆121Oct 17, 2024Updated 2 years ago
- QLoRA: Efficient Finetuning of Quantized LLMs☆11,058Jun 10, 2024Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Implementation of MaMMUT, a simple vision-encoder text-decoder architecture for multimodal tasks from Google, in Pytorch☆103Oct 10, 2023Updated 3 years ago
- Official implementation of TransNormerLLM: A Faster and Better LLM☆254Jan 23, 2024Updated 2 years ago
- [NeurIPS 2023] MeZO: Fine-Tuning Language Models with Just Forward Passes. https://arxiv.org/abs/2305.17333☆1,179Jan 11, 2024Updated 2 years ago
- Implementation of SoundStorm, Efficient Parallel Audio Generation from Google Deepmind, in Pytorch☆1,546Apr 24, 2025Updated last year
- Implementation of Mega, the Single-head Attention with Multi-headed EMA architecture that currently holds SOTA on Long Range Arena☆206Aug 26, 2023Updated 3 years ago
- Implementation of AudioLM, a SOTA Language Modeling Approach to Audio Generation out of Google Research, in Pytorch☆2,627Jan 12, 2025Updated last year
- Foundation Architecture for (M)LLMs☆3,143Apr 11, 2024Updated 2 years ago
- Standalone Product Key Memory module in Pytorch - for augmenting Transformer models☆89Nov 1, 2025Updated 11 months ago
- 🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch☆2,206Jul 9, 2026Updated 3 months ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- Sequence modeling with Mega.☆300Jan 28, 2023Updated 3 years ago
- Implementation of Block Recurrent Transformer - Pytorch☆226Aug 20, 2024Updated 2 years ago
- Fast and memory-efficient exact attention☆25,221Updated this week
- ImageBind One Embedding Space to Bind Them All☆9,080Nov 21, 2025Updated 11 months ago
- [ICLR 2024] Fine-tuning LLaMA to follow Instructions within 1 Hour and 1.2M Parameters☆5,912Mar 14, 2024Updated 2 years ago
- [ICML 2024] Break the Sequential Dependency of LLM Inference Using Lookahead Decoding☆1,344Mar 6, 2025Updated last year
- [ICLR 2024] Efficient Streaming Language Models with Attention Sinks☆7,273Jul 11, 2024Updated 2 years ago
- An Open-source Streaming High-fidelity Neural Audio Codec☆514Mar 4, 2025Updated last year
- Mamba SSM architecture☆18,960Jul 22, 2026Updated 3 months ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Implementation of N-Grammer, augmenting Transformers with latent n-grams, in Pytorch☆82Dec 4, 2022Updated 3 years ago
- The repository for the code of the UltraFastBERT paper☆518Mar 24, 2024Updated 2 years ago
- Implementation of a Transformer, but completely in Triton☆279Apr 5, 2022Updated 4 years ago
- [IJCAI'23] Learning to Speak from Text for Low-Resource TTS☆65May 30, 2023Updated 3 years ago
- Implementation of an Attention layer where each head can attend to more than just one token, using coordinate descent to pick topk☆47Jul 16, 2023Updated 3 years ago
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆221Aug 24, 2026Updated 2 months ago
- Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities☆22,234Sep 21, 2026Updated last month