NVIDIA / atexLinks
A TensorFlow Extension: GPU performance tools for TensorFlow.
☆26Updated 2 years ago
Alternatives and similar repositories for atex
Users that are interested in atex are comparing it to the libraries listed below
Sorting:
- Benchmarks to capture important workloads.☆31Updated 6 months ago
- PyTorch distributed training acceleration framework☆51Updated last week
- AI Accelerator Benchmark focuses on evaluating AI Accelerators from a practical production perspective, including the ease of use and ver…☆257Updated this week
- Training material for Nsight developer tools☆163Updated last year
- ☆31Updated 6 months ago
- Standalone Flash Attention v2 kernel without libtorch dependency☆111Updated 11 months ago
- A GPU-driven system framework for scalable AI applications☆117Updated 6 months ago
- ☆50Updated last year
- HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of…☆165Updated this week
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆348Updated this week
- ☆84Updated last week
- A Python library transfers PyTorch tensors between CPU and NVMe☆120Updated 8 months ago
- An unofficial cuda assembler, for all generations of SASS, hopefully :)☆83Updated 2 years ago
- An IR for efficiently simulating distributed ML computation.☆29Updated last year
- ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.☆100Updated 3 months ago
- An extension library of WMMA API (Tensor Core API)☆99Updated last year
- ☆150Updated 7 months ago
- ☆14Updated 3 years ago
- MLPerf™ logging library☆36Updated this week
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆40Updated 5 months ago
- ☆74Updated 4 months ago
- ☆23Updated 7 months ago
- ☆97Updated 11 months ago
- Fast and memory-efficient exact attention☆87Updated last week
- A tool for bandwidth measurements on NVIDIA GPUs.☆515Updated 4 months ago
- Compiler Infrastructure for Neural Networks☆147Updated 2 years ago
- oneAPI Collective Communications Library (oneCCL)☆241Updated 2 weeks ago
- ☆27Updated 6 months ago
- OneFlow Serving☆20Updated 4 months ago
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆94Updated 2 weeks ago