kailums / flash-attention-rocmLinks
Fast and memory-efficient exact attention ported to rocm
☆12Updated 2 years ago
Alternatives and similar repositories for flash-attention-rocm
Users that are interested in flash-attention-rocm are comparing it to the libraries listed below
Sorting:
- Implementation of IceFormer: Accelerated Inference with Long-Sequence Transformers on CPUs (ICLR 2024).☆25Updated 5 months ago
- Linear Attention Sequence Parallelism (LASP)☆88Updated last year
- A collection of reproducible inference engine benchmarks☆38Updated 8 months ago
- ☆12Updated 7 months ago
- Repository for CPU Kernel Generation for LLM Inference☆27Updated 2 years ago
- ☆78Updated last year
- ☆16Updated 2 years ago
- A library for simplifying training with multi gpu setups in the HuggingFace / PyTorch ecosystem.☆16Updated 2 weeks ago
- Implementation of the paper: "Leave No Context Behind: Efficient Infinite Context Transformers with Infini-attention" from Google in pyTO…☆58Updated this week
- ☆59Updated 2 years ago
- ☆43Updated 4 months ago
- ☆17Updated last year
- A public implementation of the ReLoRA pretraining method, built on Lightning-AI's Pytorch Lightning suite.☆35Updated last year
- Fast LLM Training CodeBase With dynamic strategy choosing [Deepspeed+Megatron+FlashAttention+CudaFusionKernel+Compiler];☆41Updated last year
- OLMost every training recipe you need to perform data interventions with the OLMo family of models.☆62Updated this week
- Cookbook of SGLang - Recipe☆45Updated this week
- NeurIPS 2023 - Cappy: Outperforming and Boosting Large Multi-Task LMs with a Small Scorer☆45Updated last year
- This is a new metric that can be used to evaluate faithfulness of text generated by LLMs. The work behind this repository can be found he…☆31Updated 2 years ago
- [ACL 2024] RelayAttention for Efficient Large Language Model Serving with Long System Prompts☆40Updated last year
- Implementation of the model: "Reka Core, Flash, and Edge: A Series of Powerful Multimodal Language Models" in PyTorch☆29Updated this week
- My Implementation of Q-Sparse: All Large Language Models can be Fully Sparsely-Activated☆33Updated last year
- Tooling for exact and MinHash deduplication of large-scale text datasets☆46Updated last week
- Implementation of the Mamba SSM with hf_integration.☆56Updated last year
- The open source implementation of "Connecting Large Language Models with Evolutionary Algorithms Yields Powerful Prompt Optimizers"☆19Updated last year
- Repository for Sparse Finetuning of LLMs via modified version of the MosaicML llmfoundry☆42Updated last year
- code for paper "Accessing higher dimensions for unsupervised word translation"☆22Updated 2 years ago
- Multi-Layer Key-Value sharing experiments on Pythia models☆34Updated last year
- Large Scale Distributed Model Training strategy with Colossal AI and Lightning AI☆56Updated 2 years ago
- Embroid: Unsupervised Prediction Smoothing Can Improve Few-Shot Classification☆11Updated 2 years ago
- Make triton easier☆49Updated last year