nod-ai / SRT
Nod.ai π¦ version of π» . You probably want to start at https://github.com/nod-ai/shark for the product and the upstream IREE repository for mainline development. This repository houses branches and configuration that aren't ready for commit upstream.
β106Updated 2 months ago
Alternatives and similar repositories for SRT:
Users that are interested in SRT are comparing it to the libraries listed below
- benchmarking some transformer deploymentsβ26Updated 2 years ago
- Home for OctoML PyTorch Profilerβ108Updated last year
- β51Updated 7 months ago
- Customized matrix multiplication kernelsβ54Updated 3 years ago
- β13Updated 3 years ago
- Training neural networks in TensorFlow 2.0 with 5x less memoryβ129Updated 3 years ago
- torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters iβ¦β178Updated 3 months ago
- Unified compiler/runtime for interfacing with PyTorch Dynamo.β99Updated last month
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.β255Updated this week
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mindβ¦β155Updated 3 months ago
- An experimental CPU backend for Triton (https//github.com/openai/triton)β40Updated 2 weeks ago
- MLIR-based partitioning systemβ76Updated this week
- Benchmarks to capture important workloads.β30Updated 2 months ago
- Fast sparse deep learning on CPUsβ53Updated 2 years ago
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")β313Updated this week
- PyTorch interface for the IPUβ177Updated last year
- A Data-Centric Compiler for Machine Learningβ82Updated last year
- An open-source efficient deep learning framework/compiler, written in python.β692Updated last month
- extensible collectives library in tritonβ84Updated this week
- β49Updated last year
- High-Performance SGEMM on CUDA devicesβ88Updated 2 months ago
- A tensor-aware point-to-point communication primitive for machine learningβ255Updated 2 years ago
- A thin, highly portable toolkit for efficiently compiling dense loop-based computation.β148Updated 2 years ago
- 3X speedup over Appleβs TensorFlow plugin by using Apache TVM on M1β136Updated 3 years ago
- Backward compatible ML compute opset inspired by HLO/MHLOβ457Updated last week
- IREE's PyTorch Frontend, based on Torch Dynamo.β74Updated this week
- Python bindings for UCXβ126Updated last week
- β85Updated 2 weeks ago
- A library of GPU kernels for sparse matrix operations.β259Updated 4 years ago
- PyTorch RFCs (experimental)β131Updated 7 months ago