pytorch-tpu / llamaLinks
Inference code for LLaMA models
☆21Updated 5 months ago
Alternatives and similar repositories for llama
Users that are interested in llama are comparing it to the libraries listed below
Sorting:
- Examples for MS-AMP package.☆30Updated 2 months ago
- ☆119Updated last year
- Boosting 4-bit inference kernels with 2:4 Sparsity☆82Updated last year
- Easy and Efficient Quantization for Transformers☆203Updated 2 months ago
- Ditto is an open-source framework that enables direct conversion of HuggingFace PreTrainedModels into TensorRT-LLM engines.☆49Updated 2 months ago
- ☆159Updated 2 years ago
- QUICK: Quantization-aware Interleaving and Conflict-free Kernel for efficient LLM inference☆118Updated last year
- ☆27Updated last year
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆216Updated last year
- ☆121Updated last year
- ☆112Updated last year
- Research and development for optimizing transformers☆130Updated 4 years ago
- Implementation of a Transformer, but completely in Triton☆274Updated 3 years ago
- Simple implementation of Speculative Sampling in NumPy for GPT-2.☆96Updated 2 years ago
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆212Updated this week
- Benchmark code for the "Online normalizer calculation for softmax" paper☆99Updated 7 years ago
- [NeurIPS'23] Speculative Decoding with Big Little Decoder☆94Updated last year
- Applied AI experiments and examples for PyTorch☆294Updated last month
- PyTorch bindings for CUTLASS grouped GEMM.☆119Updated 3 months ago
- This repository contains the experimental PyTorch native float8 training UX☆224Updated last year
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆265Updated 2 months ago
- A collection of memory efficient attention operators implemented in the Triton language.☆278Updated last year
- The official implementation of the EMNLP 2023 paper LLM-FP4☆214Updated last year
- ☆74Updated 5 months ago
- GPTQ inference Triton kernel☆307Updated 2 years ago
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆114Updated last year
- ☆237Updated this week
- Faster Pytorch bitsandbytes 4bit fp4 nn.Linear ops☆30Updated last year
- extensible collectives library in triton☆87Updated 5 months ago
- Fast sparse deep learning on CPUs☆56Updated 2 years ago