pytorch-tpu / llama
Inference code for LLaMA models
☆21Updated last month
Alternatives and similar repositories for llama
Users that are interested in llama are comparing it to the libraries listed below
Sorting:
- Best practices for testing advanced Mixtral, DeepSeek, and Qwen series MoE models using Megatron Core MoE.☆10Updated 3 weeks ago
- ☆158Updated last year
- Boosting 4-bit inference kernels with 2:4 Sparsity☆73Updated 8 months ago
- Examples for MS-AMP package.☆29Updated last year
- Odysseus: Playground of LLM Sequence Parallelism☆69Updated 11 months ago
- ☆104Updated 8 months ago
- QUICK: Quantization-aware Interleaving and Conflict-free Kernel for efficient LLM inference☆117Updated last year
- ☆106Updated 11 months ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆124Updated this week
- ☆26Updated last year
- ☆69Updated last month
- Benchmark code for the "Online normalizer calculation for softmax" paper☆92Updated 6 years ago
- Applied AI experiments and examples for PyTorch☆267Updated this week
- [NeurIPS'23] Speculative Decoding with Big Little Decoder☆92Updated last year
- ☆79Updated 6 months ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆209Updated 8 months ago
- Repository for Sparse Finetuning of LLMs via modified version of the MosaicML llmfoundry☆41Updated last year
- llama INT4 cuda inference with AWQ☆54Updated 3 months ago
- System for automated integration of deep learning backends.☆48Updated 2 years ago
- extensible collectives library in triton☆86Updated last month
- Research and development for optimizing transformers☆126Updated 4 years ago
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆109Updated 10 months ago
- Simple implementation of Speculative Sampling in NumPy for GPT-2.☆95Updated last year
- ☆55Updated last year
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆248Updated 6 months ago
- pytorch-profiler☆51Updated last year
- ☆76Updated 4 months ago
- ☆117Updated last year
- ☆203Updated 10 months ago
- ☆20Updated last month