kotoba-tech / kotomambaView external linksLinks
Mamba training library developed by kotoba technologies
☆71Feb 11, 2024Updated 2 years ago
Alternatives and similar repositories for kotomamba
Users that are interested in kotomamba are comparing it to the libraries listed below
Sorting:
- ☆24Sep 25, 2024Updated last year
- APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…☆30Jan 28, 2026Updated 2 weeks ago
- Support Continual pre-training & Instruction Tuning forked from llama-recipes☆34Feb 17, 2024Updated last year
- ☆51Jan 28, 2024Updated 2 years ago
- ☆45Apr 30, 2018Updated 7 years ago
- Example of using Epochraft to train HuggingFace transformers models with PyTorch FSDP☆11Jan 29, 2024Updated 2 years ago
- Reference implementation of "Softmax Attention with Constant Cost per Token" (Heinsen, 2024)☆24Jun 6, 2024Updated last year
- ☆35Nov 22, 2024Updated last year
- ☆16Dec 19, 2024Updated last year
- ☆19Dec 6, 2024Updated last year
- FlexAttention w/ FlashAttention3 Support☆27Oct 5, 2024Updated last year
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32May 25, 2024Updated last year
- Griffin MQA + Hawk Linear RNN Hybrid☆88Apr 26, 2024Updated last year
- Implementation of 'Vocos: Closing the gap between time-domain and Fourier-based neural vocoders for high-quality audio synthesis', in MLX☆23Oct 30, 2024Updated last year
- ☆35Apr 12, 2024Updated last year
- ☆20May 30, 2024Updated last year
- HGRN2: Gated Linear RNNs with State Expansion☆56Aug 20, 2024Updated last year
- [NeurIPS 2023] Sparse Modular Activation for Efficient Sequence Modeling☆40Dec 2, 2023Updated 2 years ago
- ☆19Dec 4, 2025Updated 2 months ago
- AGaLiTe: Approximate Gated Linear Transformers for Online Reinforcement Learning (Published in TMLR)☆23Oct 15, 2024Updated last year
- Here we will test various linear attention designs.☆62Apr 25, 2024Updated last year
- JAX/Flax implementation of the Hyena Hierarchy☆34Apr 27, 2023Updated 2 years ago
- 日本十進分類法のIME辞書☆11Dec 8, 2022Updated 3 years ago
- ☆82Apr 16, 2024Updated last year
- Easily turn large English text datasets into Japanese text datasets using open LLMs.☆25Jan 20, 2025Updated last year
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆93Jan 25, 2024Updated 2 years ago
- Accelerated First Order Parallel Associative Scan☆196Jan 7, 2026Updated last month
- ☆22Dec 15, 2023Updated 2 years ago
- Why Do We Need Weight Decay in Modern Deep Learning? [NeurIPS 2024]☆70Sep 25, 2024Updated last year
- possibly useful materials for learning RWKV language model.☆26Jun 8, 2023Updated 2 years ago
- Implementation of Hyena Hierarchy in JAX☆10Apr 30, 2023Updated 2 years ago
- PyTorch implementation for PaLM: A Hybrid Parser and Language Model.☆10Jan 7, 2020Updated 6 years ago
- ☆12Feb 27, 2022Updated 3 years ago
- Embroid: Unsupervised Prediction Smoothing Can Improve Few-Shot Classification☆11Aug 12, 2023Updated 2 years ago
- ☆11Oct 11, 2023Updated 2 years ago
- Advanced Formal Language Theory (263-5352-00L; Frühjahr 2023)☆10Feb 21, 2023Updated 2 years ago
- ☆10May 16, 2024Updated last year
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"☆248Jun 6, 2025Updated 8 months ago
- ☆29Jul 9, 2024Updated last year