kailums / flash-attention-rocmLinks
Fast and memory-efficient exact attention ported to rocm
☆11Updated last year
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 last week
- 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 last year
- Implementation of the paper: "Leave No Context Behind: Efficient Infinite Context Transformers with Infini-attention" from Google in pyTO…☆55Updated 3 weeks ago
- code for paper "Accessing higher dimensions for unsupervised word translation"☆21Updated 2 years ago
- The open source implementation of "Connecting Large Language Models with Evolutionary Algorithms Yields Powerful Prompt Optimizers"☆19Updated last year
- A collection of reproducible inference engine benchmarks☆32Updated 3 months ago
- Implementation of the model: "Reka Core, Flash, and Edge: A Series of Powerful Multimodal Language Models" in PyTorch☆30Updated this week
- A GPT-based generative LM for combined text and math formulas, leveraging tree-based formula encoding.☆40Updated 2 years ago
- Minimum Description Length probing for neural network representations☆18Updated 5 months ago
- ☆55Updated last week
- Large Scale Distributed Model Training strategy with Colossal AI and Lightning AI☆57Updated last year
- Fast LLM Training CodeBase With dynamic strategy choosing [Deepspeed+Megatron+FlashAttention+CudaFusionKernel+Compiler];☆40Updated last year
- setup the env for vllm users☆16Updated last year
- ☆76Updated 7 months ago
- A Python implementation of Toolformer using Huggingface Transformers☆14Updated 2 years ago
- ☆16Updated last year
- Implementation of "LM-Infinite: Simple On-the-Fly Length Generalization for Large Language Models"☆40Updated 8 months ago
- Repository for CPU Kernel Generation for LLM Inference☆26Updated 2 years ago
- Make triton easier☆47Updated last year
- PyTorch Implementation of the paper "MM1: Methods, Analysis & Insights from Multimodal LLM Pre-training"☆24Updated 3 weeks ago
- Library for the Test-based Calibration Error (TCE) metric to quantify the degree to classifier calibration.☆13Updated last year
- Awesome Triton Resources☆32Updated 2 months ago
- ☆51Updated last year
- Embroid: Unsupervised Prediction Smoothing Can Improve Few-Shot Classification☆11Updated last year
- Linear Attention Sequence Parallelism (LASP)☆85Updated last year
- Contextual Position Encoding but with some custom CUDA Kernels https://arxiv.org/abs/2405.18719☆22Updated last year
- ☆48Updated 10 months ago
- Evaluation of bm42 sparse indexing algorithm☆68Updated last year
- ☆26Updated 2 years ago
- KITE (Knowledge-Intensive Task Evaluation) is an end-to-end benchmark for RAG pipelines☆19Updated 11 months ago