pytorch-tpu / llamaLinks
Inference code for LLaMA models
☆21Updated 2 months ago
Alternatives and similar repositories for llama
Users that are interested in llama are comparing it to the libraries listed below
Sorting:
- ☆105Updated 9 months ago
- ☆108Updated last year
- ☆157Updated last year
- Boosting 4-bit inference kernels with 2:4 Sparsity☆76Updated 9 months ago
- ☆71Updated 2 months ago
- ☆26Updated last year
- Repository for Sparse Finetuning of LLMs via modified version of the MosaicML llmfoundry☆42Updated last year
- [NeurIPS'23] Speculative Decoding with Big Little Decoder☆92Updated last year
- Examples for MS-AMP package.☆29Updated last year
- QUICK: Quantization-aware Interleaving and Conflict-free Kernel for efficient LLM inference☆118Updated last year
- Simple implementation of Speculative Sampling in NumPy for GPT-2.☆95Updated last year
- This repository contains the experimental PyTorch native float8 training UX☆223Updated 10 months ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆210Updated 9 months ago
- Best practices for testing advanced Mixtral, DeepSeek, and Qwen series MoE models using Megatron Core MoE.☆17Updated this week
- Faster Pytorch bitsandbytes 4bit fp4 nn.Linear ops☆28Updated last year
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆127Updated this week
- extensible collectives library in triton☆87Updated 2 months ago
- ☆59Updated last year
- Fast sparse deep learning on CPUs☆53Updated 2 years ago
- Applied AI experiments and examples for PyTorch☆274Updated last week
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆251Updated 7 months ago
- Odysseus: Playground of LLM Sequence Parallelism☆70Updated 11 months ago
- ☆119Updated last year
- Research and development for optimizing transformers☆126Updated 4 years ago
- Benchmark code for the "Online normalizer calculation for softmax" paper☆94Updated 6 years ago
- ☆144Updated 2 years ago
- PyTorch bindings for CUTLASS grouped GEMM.☆93Updated last week
- Implementation of a Transformer, but completely in Triton☆266Updated 3 years ago
- PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"☆60Updated 2 months ago
- pytorch-profiler☆51Updated 2 years ago