ROCm / pyrsmi
python package of rocm-smi-lib
☆20Updated 7 months ago
Alternatives and similar repositories for pyrsmi
Users that are interested in pyrsmi are comparing it to the libraries listed below
Sorting:
- ☆68Updated last month
- extensible collectives library in triton☆86Updated last month
- ☆21Updated 2 months ago
- ☆32Updated this week
- Write a fast kernel and run it on Discord. See how you compare against the best!☆44Updated last week
- Ahead of Time (AOT) Triton Math Library☆63Updated this week
- Framework to reduce autotune overhead to zero for well known deployments.☆70Updated this week
- Intel® Extension for DeepSpeed* is an extension to DeepSpeed that brings feature support with SYCL kernels on Intel GPU(XPU) device. Note…☆61Updated 2 months ago
- oneCCL Bindings for Pytorch*☆97Updated 2 weeks ago
- A Python library transfers PyTorch tensors between CPU and NVMe☆115Updated 5 months ago
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆106Updated 10 months ago
- An experimental CPU backend for Triton (https//github.com/openai/triton)☆40Updated last month
- High-Performance SGEMM on CUDA devices☆91Updated 3 months ago
- Benchmarks to capture important workloads.☆31Updated 3 months ago
- Evaluating Large Language Models for CUDA Code Generation ComputeEval is a framework designed to generate and evaluate CUDA code from Lar…☆38Updated 2 weeks ago
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆84Updated this week
- ☆79Updated 6 months ago
- Test suite for probing the numerical behavior of NVIDIA tensor cores☆38Updated 9 months ago
- ☆104Updated 8 months ago
- Load compute kernels from the Hub☆116Updated this week
- Boosting 4-bit inference kernels with 2:4 Sparsity☆73Updated 8 months ago
- Explore training for quantized models☆18Updated 4 months ago
- Microsoft Collective Communication Library☆65Updated 5 months ago
- Applied AI experiments and examples for PyTorch☆265Updated 2 weeks ago
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆45Updated 9 months ago
- Repository for CPU Kernel Generation for LLM Inference☆26Updated last year
- Experiment of using Tangent to autodiff triton☆78Updated last year
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆132Updated this week
- Intel Gaudi's Megatron DeepSpeed Large Language Models for training☆13Updated 4 months ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆124Updated this week