AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inference.
☆12Jun 24, 2024Updated last year
Alternatives and similar repositories for AITemplate
Users that are interested in AITemplate are comparing it to the libraries listed below
Sorting:
- benchmarking miopen☆17Jan 14, 2019Updated 7 years ago
- a simple Flash Attention v2 implementation with ROCM (RDNA3 GPU, roc wmma), mainly used for stable diffusion(ComfyUI) in Windows ZLUDA en…☆51Aug 25, 2024Updated last year
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆154Jan 21, 2026Updated last month
- Guides to hopefully simplify the process of using ROCm.☆12Sep 26, 2024Updated last year
- access ChatGPT/Gemini/Claude from Emacs without APIs☆10Dec 25, 2025Updated 2 months ago
- ☆12Aug 26, 2025Updated 6 months ago
- 1st Place Team Crane: @aswinkumar1999 @rathull @kyolebu☆29Sep 8, 2025Updated 5 months ago
- THIS REPOSITORY HAS MOVED TO github.com/nvidia/cub, WHICH IS AUTOMATICALLY MIRRORED HERE.☆11May 6, 2023Updated 2 years ago
- GPU implementation of Winograd convolution☆10Oct 23, 2017Updated 8 years ago
- Templates for commonly used GitHub actions steps☆13Dec 13, 2024Updated last year
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆55Jan 22, 2026Updated last month
- A human-friendly implementation of the iRobot Open Interface version 2 API.☆14May 14, 2016Updated 9 years ago
- A tool for visualization of complex job searches.☆13Jul 8, 2022Updated 3 years ago
- KANs and MLPs☆12Jun 7, 2024Updated last year
- Notes on putting micropython on STM32F407VG bare board☆11Oct 7, 2019Updated 6 years ago
- Example of binding a TF32 CUTLASS GEMM kernel to PyTorch☆12Jun 7, 2024Updated last year
- ☆15Mar 26, 2025Updated 11 months ago
- "Not too complicated" training code for CIFAR-10 by PyTorch Lightning☆12Jun 5, 2022Updated 3 years ago
- ☆16Jun 15, 2023Updated 2 years ago
- ☆15Aug 3, 2021Updated 4 years ago
- hipDF - GPU DataFrame Library☆16Dec 19, 2025Updated 2 months ago
- Accelerating CNN's convolution operation on GPUs by using memory-efficient data access patterns.☆14Dec 8, 2017Updated 8 years ago
- Simple repository contribution statistics☆15Updated this week
- N-body simulation based on CUDA.☆14Jun 20, 2019Updated 6 years ago
- Running PyTorch on Windows with AMD GPUs using alpha ROCm wheels. It's fast, it's fragile, and it hates you back.☆23Oct 31, 2025Updated 4 months ago
- [DEPRECATED] Moved to ROCm/rocm-libraries repo. NOTE: develop branch is maintained as a read-only mirror☆523Updated this week
- Triton Server Component for lightning.ai☆14Feb 15, 2023Updated 3 years ago
- A scraper for downloading free Springer e-books.☆15May 1, 2020Updated 5 years ago
- Collection of scripts to build PyTorch and the domain libraries from source.☆13Feb 4, 2026Updated last month
- 该部分为自己在学习tensorflow2.0中实现的各种模型还有算法，供大家参考☆20Jul 30, 2020Updated 5 years ago
- Inference Llama 2 with a model compiled to native code by TorchInductor☆14Feb 8, 2024Updated 2 years ago
- Print github contribution☆14Apr 19, 2023Updated 2 years ago
- add HTML links and targets to the tags within Sphinx Python docstrings☆19Nov 27, 2024Updated last year
- CUDA keyring packaging for Debian☆14Apr 14, 2023Updated 2 years ago
- ☆16Jan 23, 2025Updated last year
- ☆16Nov 19, 2025Updated 3 months ago
- Utility for merging multiple XUnit xml reports into a single xml report.☆15Jul 3, 2024Updated last year
- A directive for including a Plotly figure in a Sphinx document.☆16Oct 25, 2022Updated 3 years ago
- Python SDK for TraceRoot☆37Nov 12, 2025Updated 3 months ago