juvi21 / CoPE-cuda
Contextual Position Encoding but with some custom CUDA Kernels https://arxiv.org/abs/2405.18719
☆22Updated 11 months ago
Alternatives and similar repositories for CoPE-cuda:
Users that are interested in CoPE-cuda are comparing it to the libraries listed below
- ☆53Updated 10 months ago
- Code for preprint "Metadata Conditioning Accelerates Language Model Pre-training (MeCo)"☆37Updated this week
- ☆20Updated 11 months ago
- Implementation of the paper: "Leave No Context Behind: Efficient Infinite Context Transformers with Infini-attention" from Google in pyTO…☆55Updated 2 weeks ago
- Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"☆27Updated last year
- ☆45Updated 2 months ago
- GoldFinch and other hybrid transformer components☆45Updated 9 months ago
- An Experiment on Dynamic NTK Scaling RoPE☆64Updated last year
- triton ver of gqa flash attn, based on the tutorial☆11Updated 9 months ago
- Official repository for the paper "Approximating Two-Layer Feedforward Networks for Efficient Transformers"☆37Updated last year
- Implementation of the model: "Reka Core, Flash, and Edge: A Series of Powerful Multimodal Language Models" in PyTorch☆30Updated 2 weeks ago
- ☆47Updated last year
- Official implementation of "The Sparse Frontier: Sparse Attention Trade-offs in Transformer LLMs"☆26Updated 2 weeks ago
- Linear Attention Sequence Parallelism (LASP)☆82Updated 11 months ago
- ☆22Updated last year
- ☆14Updated 2 years ago
- [ICLR 2024] CLEX: Continuous Length Extrapolation for Large Language Models☆76Updated last year
- ☆24Updated last year
- DPO, but faster 🚀☆42Updated 5 months ago
- Open-sourcing code associated with the AAAI-25 paper "On the Expressiveness and Length Generalization of Selective State-Space Models on …☆13Updated this week
- ☆19Updated 2 years ago
- [NeurIPS 2023] Sparse Modular Activation for Efficient Sequence Modeling☆36Updated last year
- ☆30Updated 11 months ago
- The source code of our work "Prepacking: A Simple Method for Fast Prefilling and Increased Throughput in Large Language Models" [AISTATS …☆59Updated 7 months ago
- ☆64Updated last year
- Repository for CPU Kernel Generation for LLM Inference☆26Updated last year
- [ICML'24] The official implementation of “Rethinking Optimization and Architecture for Tiny Language Models”☆121Updated 3 months ago
- Script for processing OpenAI's PRM800K process supervision dataset into an Alpaca-style instruction-response format☆27Updated last year
- Code for paper "Patch-Level Training for Large Language Models"☆84Updated 5 months ago
- Transformers at any scale☆41Updated last year