nod-ai / SRTLinks
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 9 months ago
Alternatives and similar repositories for SRT
Users that are interested in SRT are comparing it to the libraries listed below
Sorting:
- benchmarking some transformer deploymentsβ26Updated 2 years ago
- 3X speedup over Appleβs TensorFlow plugin by using Apache TVM on M1β138Updated 3 years ago
- A tracing JIT compiler for PyTorchβ13Updated 3 years ago
- PyTorch interface for the IPUβ181Updated last year
- torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters iβ¦β181Updated last month
- Customized matrix multiplication kernelsβ57Updated 3 years ago
- A thin, highly portable toolkit for efficiently compiling dense loop-based computation.β148Updated 2 years ago
- PyTorch RFCs (experimental)β135Updated 4 months ago
- Home for OctoML PyTorch Profilerβ114Updated 2 years ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.β122Updated 3 weeks ago
- β52Updated last year
- GPU implementation of a fast generalized ANS (asymmetric numeral system) entropy encoder and decoder, with extensions for lossless compreβ¦β353Updated 3 months ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.β301Updated this week
- Training neural networks in TensorFlow 2.0 with 5x less memoryβ136Updated 3 years ago
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mindβ¦β161Updated 3 weeks ago
- π Interactive performance profiling and debugging tool for PyTorch neural networks.β64Updated 8 months ago
- Benchmarks to capture important workloads.β31Updated 8 months ago
- An open-source efficient deep learning framework/compiler, written in python.β731Updated last month
- β333Updated 3 weeks ago
- Torch Distributed Experimentalβ117Updated last year
- Repository for the QUIK project, enabling the use of 4bit kernels for generative inference - EMNLP 2024β182Updated last year
- Fast sparse deep learning on CPUsβ56Updated 3 years ago
- The Foundation for All Legate Librariesβ228Updated last week
- β74Updated last year
- Notes and artifacts from the ONNX steering committeeβ26Updated last week
- Accelerate PyTorch models with ONNX Runtimeβ364Updated 7 months ago
- Implementation of Flash Attention in Jaxβ219Updated last year
- Example code and applications for machine learning on Graphcore IPUsβ330Updated last year
- Productionize machine learning predictions, with ONNX or withoutβ66Updated last year
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.β45Updated last year