pytorch-labs / applied-ai
Applied AI experiments and examples for PyTorch
☆262Updated last week
Alternatives and similar repositories for applied-ai:
Users that are interested in applied-ai are comparing it to the libraries listed below
- Fast low-bit matmul kernels in Triton☆295Updated this week
- Cataloging released Triton kernels.☆220Updated 3 months ago
- Collection of kernels written in Triton language☆120Updated last month
- This repository contains the experimental PyTorch native float8 training UX☆224Updated 9 months ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆120Updated this week
- ☆202Updated last week
- ☆202Updated 9 months ago
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆194Updated this week
- ☆104Updated 8 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆87Updated this week
- ☆70Updated 4 months ago
- A curated collection of resources, tutorials, and best practices for learning and mastering NVIDIA CUTLASS☆169Updated last month
- Perplexity GPU Kernels☆272Updated this week
- extensible collectives library in triton☆85Updated last month
- ☆78Updated 5 months ago
- A collection of memory efficient attention operators implemented in the Triton language.☆266Updated 11 months ago
- Fastest kernels written from scratch☆252Updated last month
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆248Updated 6 months ago
- Fast Hadamard transform in CUDA, with a PyTorch interface☆183Updated 11 months ago
- KernelBench: Can LLMs Write GPU Kernels? - Benchmark with Torch -> CUDA problems☆288Updated last week
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆106Updated 9 months ago
- Triton-based implementation of Sparse Mixture of Experts.☆212Updated 5 months ago
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆244Updated this week
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆209Updated 8 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆120Updated 4 months ago
- Code for Neurips24 paper: QuaRot, an end-to-end 4-bit inference of large language models.☆383Updated 5 months ago
- ring-attention experiments☆132Updated 6 months ago
- [MLSys'24] Atom: Low-bit Quantization for Efficient and Accurate LLM Serving☆307Updated 10 months ago
- [NeurIPS 2024] KVQuant: Towards 10 Million Context Length LLM Inference with KV Cache Quantization☆346Updated 8 months ago
- ☆157Updated last year