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.
TransferBench is a utility capable of benchmarking simultaneous copies between user-specified devices (CPUs/GPUs)
PyTorch Extension Library of Optimized Scatter Operations
Additional utils and helpers to extend TensorFlow when build recommendation systems, contributed and maintained by SIG Recommenders.
The lightweight PyTorch wrapper for high-performance AI research. Scale your models, not the boilerplate.
最近更新: 1天前Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators
最近更新: 1天前Data manipulation and transformation for audio signal processing, powered by PyTorch
最近更新: 1天前