juvi21 / CoPE-cuda
Contextual Position Encoding but with some custom CUDA Kernels https://arxiv.org/abs/2405.18719
☆22Updated 10 months ago
Alternatives and similar repositories for CoPE-cuda:
Users that are interested in CoPE-cuda are comparing it to the libraries listed below
- Code for preprint "Metadata Conditioning Accelerates Language Model Pre-training (MeCo)"☆37Updated 3 weeks ago
- ☆53Updated 9 months ago
- ☆19Updated 10 months ago
- triton ver of gqa flash attn, based on the tutorial☆11Updated 8 months ago
- An Experiment on Dynamic NTK Scaling RoPE☆62Updated last year
- ☆30Updated 10 months ago
- ☆64Updated last year
- Linear Attention Sequence Parallelism (LASP)☆81Updated 10 months ago
- Transformers at any scale☆41Updated last year
- Script for processing OpenAI's PRM800K process supervision dataset into an Alpaca-style instruction-response format☆27Updated last year
- Exploration of automated dataset selection approaches at large scales.☆37Updated last month
- Fast LLM Training CodeBase With dynamic strategy choosing [Deepspeed+Megatron+FlashAttention+CudaFusionKernel+Compiler];☆36Updated last year
- 32 times longer context window than vanilla Transformers and up to 4 times longer than memory efficient Transformers.☆47Updated last year
- ☆20Updated this week
- Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"☆26Updated 11 months ago
- This is a personal reimplementation of Google's Infini-transformer, utilizing a small 2b model. The project includes both model and train…☆56Updated 11 months ago
- Repository for Sparse Finetuning of LLMs via modified version of the MosaicML llmfoundry☆40Updated last year
- A toolkit for scaling law research ⚖☆49Updated 2 months ago
- Train, tune, and infer Bamba model☆88Updated 3 months ago
- Here we will test various linear attention designs.☆60Updated 11 months ago
- DPO, but faster 🚀☆40Updated 4 months ago
- [ICML'24] The official implementation of “Rethinking Optimization and Architecture for Tiny Language Models”☆122Updated 3 months ago
- Implementation of the paper: "Leave No Context Behind: Efficient Infinite Context Transformers with Infini-attention" from Google in pyTO…☆55Updated last week
- [ICML'24 Oral] The official code of "DiJiang: Efficient Large Language Models through Compact Kernelization", a novel DCT-based linear at…☆100Updated 10 months ago
- Code for RL4F: Generating Natural Language Feedback with Reinforcement Learning for Repairing Model Outputs. ACL 2023.☆63Updated 4 months ago
- Implementation of the model: "Reka Core, Flash, and Edge: A Series of Powerful Multimodal Language Models" in PyTorch☆30Updated this week
- GoldFinch and other hybrid transformer components☆45Updated 8 months ago
- Using FlexAttention to compute attention with different masking patterns☆43Updated 6 months ago
- This repo is based on https://github.com/jiaweizzhao/GaLore☆26Updated 6 months ago
- ☆48Updated last year