ARM-software / kleidiaiLinks
This repository is a read-only mirror of https://gitlab.arm.com/kleidi/kleidiai
☆113Updated this week
Alternatives and similar repositories for kleidiai
Users that are interested in kleidiai are comparing it to the libraries listed below
Sorting:
- An experimental CPU backend for Triton☆173Updated 2 months ago
- OpenAI Triton backend for Intel® GPUs☆225Updated last week
- BitBLAS is a library to support mixed-precision matrix multiplications, especially for quantized LLM deployment.☆749Updated 5 months ago
- FlagTree is a unified compiler supporting multiple AI chip backends for custom Deep Learning operations, which is forked from triton-lang…☆197Updated this week
- Open ABI and FFI for Machine Learning Systems☆313Updated this week
- ☆171Updated last week
- AI Tensor Engine for ROCm☆348Updated this week
- Shared Middle-Layer for Triton Compilation☆324Updated last month
- CUDA Matrix Multiplication Optimization☆256Updated last year
- We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …☆192Updated last year
- ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.☆148Updated 8 months ago
- Fast and memory-efficient exact attention☆110Updated last week
- Tile-Based Runtime for Ultra-Low-Latency LLM Inference☆557Updated last week
- PyTorch emulation library for Microscaling (MX)-compatible data formats☆337Updated 7 months ago
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆127Updated last year
- High Performance LLM Inference Operator Library☆603Updated this week
- ☆102Updated last year
- NVIDIA NVSHMEM is a parallel programming interface for NVIDIA GPUs based on OpenSHMEM. NVSHMEM can significantly reduce multi-process com…☆459Updated last month
- A model compilation solution for various hardware☆464Updated 5 months ago
- An experimental CPU backend for Triton (https//github.com/openai/triton)☆48Updated 5 months ago
- 🤖FFPA: Extend FlashAttention-2 with Split-D, ~O(1) SRAM complexity for large headdim, 1.8x~3x↑🎉 vs SDPA EA.☆248Updated last week
- triton for dsa☆56Updated this week
- ☆221Updated last year
- Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators☆515Updated this week
- TritonParse: A Compiler Tracer, Visualizer, and Reproducer for Triton Kernels☆189Updated this week
- ☆173Updated 8 months ago
- ☆116Updated this week
- Benchmark code for the "Online normalizer calculation for softmax" paper☆105Updated 7 years ago
- ☆179Updated 2 years ago
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆276Updated 6 months ago