NVIDIA / atexLinks
A TensorFlow Extension: GPU performance tools for TensorFlow.
☆26Updated last year
Alternatives and similar repositories for atex
Users that are interested in atex are comparing it to the libraries listed below
Sorting:
- A GPU-driven system framework for scalable AI applications☆114Updated 4 months ago
- An extension library of WMMA API (Tensor Core API)☆97Updated 10 months ago
- HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of…☆147Updated 2 weeks ago
- PyTorch distributed training acceleration framework☆49Updated 3 months ago
- ROCm Tracer Callback/Activity Library for Performance tracing AMD GPUs☆83Updated last week
- ☆29Updated 4 months ago
- ☆50Updated last year
- An unofficial cuda assembler, for all generations of SASS, hopefully :)☆83Updated 2 years ago
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆331Updated this week
- Bandwidth test for ROCm☆56Updated 2 weeks ago
- A visualization tool to display TF-Grappler optimized op graph☆12Updated 2 years ago
- Compiler Infrastructure for Neural Networks☆145Updated last year
- THIS REPOSITORY HAS MOVED TO github.com/nvidia/cub, WHICH IS AUTOMATICALLY MIRRORED HERE.☆84Updated last year
- MLPerf™ logging library☆36Updated last month
- Computation using data flow graphs for scalable machine learning☆67Updated this week
- Training material for Nsight developer tools☆158Updated 10 months ago
- Standalone Flash Attention v2 kernel without libtorch dependency☆110Updated 8 months ago
- oneAPI Collective Communications Library (oneCCL)☆234Updated 2 weeks ago
- Unified compiler/runtime for interfacing with PyTorch Dynamo.☆100Updated 2 weeks ago
- ☆19Updated 2 weeks ago
- AI Accelerator Benchmark focuses on evaluating AI Accelerators from a practical production perspective, including the ease of use and ver…☆242Updated 3 weeks ago
- MLIR-based partitioning system☆87Updated this week
- ☆106Updated last month
- ☆63Updated this week
- ☆13Updated 4 years ago
- ☆194Updated 2 years ago
- OneFlow->ONNX☆43Updated 2 years ago
- ☆23Updated 2 years ago
- A library for syntactically rewriting Python programs, pronounced (sinner).☆69Updated 3 years ago
- Fairring (FAIR + Herring) is a plug-in for PyTorch that provides a process group for distributed training that outperforms NCCL at large …☆65Updated 3 years ago