pytorch-tpu / llama
Inference code for LLaMA models
☆21Updated 3 weeks ago
Alternatives and similar repositories for llama:
Users that are interested in llama are comparing it to the libraries listed below
- Boosting 4-bit inference kernels with 2:4 Sparsity☆72Updated 7 months ago
- ☆157Updated last year
- QUICK: Quantization-aware Interleaving and Conflict-free Kernel for efficient LLM inference☆117Updated last year
- Odysseus: Playground of LLM Sequence Parallelism☆68Updated 10 months ago
- ☆101Updated 10 months ago
- ☆103Updated 8 months ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆116Updated this week
- [NeurIPS'23] Speculative Decoding with Big Little Decoder☆90Updated last year
- Repository for Sparse Finetuning of LLMs via modified version of the MosaicML llmfoundry☆40Updated last year
- ☆117Updated last year
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆209Updated 8 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆84Updated 5 months ago
- ☆78Updated 5 months ago
- extensible collectives library in triton☆85Updated 3 weeks ago
- [ICML 2023] SmoothQuant: Accurate and Efficient Post-Training Quantization for Large Language Models☆11Updated last year
- An innovative method expediting LLMs via streamlined semi-autoregressive generation and draft verification.☆25Updated last week
- Examples for MS-AMP package.☆29Updated last year
- Research and development for optimizing transformers☆125Updated 4 years ago
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆106Updated 9 months ago
- ☆26Updated last year
- [MLSys'24] Atom: Low-bit Quantization for Efficient and Accurate LLM Serving☆305Updated 9 months ago
- ☆68Updated last month
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆248Updated 5 months ago
- Faster Pytorch bitsandbytes 4bit fp4 nn.Linear ops☆28Updated last year
- ☆198Updated 9 months ago
- Simple implementation of Speculative Sampling in NumPy for GPT-2.☆94Updated last year
- Benchmark code for the "Online normalizer calculation for softmax" paper☆91Updated 6 years ago
- Applied AI experiments and examples for PyTorch☆262Updated last month
- ☆122Updated 2 months ago
- Fast low-bit matmul kernels in Triton☆294Updated this week