ankan-ban / llama2.cu
Inference Llama 2 in one file of pure Cuda
☆17Updated last year
Alternatives and similar repositories for llama2.cu:
Users that are interested in llama2.cu are comparing it to the libraries listed below
- llama INT4 cuda inference with AWQ☆50Updated last month
- CUDA 8-bit Tensor Core Matrix Multiplication based on m16n16k16 WMMA API☆28Updated last year
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆97Updated 7 months ago
- Several optimization methods of half-precision general matrix vector multiplication (HGEMV) using CUDA core.☆55Updated 5 months ago
- Standalone Flash Attention v2 kernel without libtorch dependency☆104Updated 5 months ago
- ☆157Updated last year
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆35Updated 5 months ago
- ☆117Updated 10 months ago
- The official implementation of the EMNLP 2023 paper LLM-FP4☆184Updated last year
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆234Updated 3 months ago
- ☆181Updated 7 months ago
- [ICML 2023] SmoothQuant: Accurate and Efficient Post-Training Quantization for Large Language Models☆20Updated 11 months ago
- Fast low-bit matmul kernels in Triton☆238Updated this week
- ☆57Updated 3 months ago
- Advanced Ultra-Low Bitrate Compression Techniques for the LLaMA Family of LLMs☆111Updated last year
- ☆180Updated this week
- A general 2-8 bits quantization toolbox with GPTQ/AWQ/HQQ/VPTQ, and export to onnx/onnx-runtime easily.☆159Updated last week
- QQQ is an innovative and hardware-optimized W4A8 quantization solution for LLMs.☆101Updated this week
- An easy-to-use package for implementing SmoothQuant for LLMs☆92Updated 9 months ago
- ☆157Updated last week
- Explore training for quantized models☆15Updated last month
- ☆67Updated 2 months ago
- A safetensors extension to efficiently store sparse quantized tensors on disk☆77Updated this week
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆88Updated 11 months ago
- [MLSys'24] Atom: Low-bit Quantization for Efficient and Accurate LLM Serving☆295Updated 7 months ago
- Boosting 4-bit inference kernels with 2:4 Sparsity☆64Updated 5 months ago
- BitBLAS is a library to support mixed-precision matrix multiplications, especially for quantized LLM deployment.☆523Updated last week
- Repository for the QUIK project, enabling the use of 4bit kernels for generative inference - EMNLP 2024☆175Updated 10 months ago
- Official PyTorch implementation of FlatQuant: Flatness Matters for LLM Quantization☆102Updated 3 weeks ago
- Inference Llama 2 in one file of pure C & one file with CUDA☆21Updated last year