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 8 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
- A tracing JIT compiler for PyTorchβ13Updated 3 years ago
- 3X speedup over Appleβs TensorFlow plugin by using Apache TVM on M1β137Updated 3 years ago
- A thin, highly portable toolkit for efficiently compiling dense loop-based computation.β148Updated 2 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 2 weeks ago
- β52Updated last year
- Customized matrix multiplication kernelsβ56Updated 3 years ago
- β13Updated 4 years ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.β296Updated this week
- Torch Distributed Experimentalβ117Updated last year
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.β123Updated last month
- β74Updated last year
- Home for OctoML PyTorch Profilerβ114Updated 2 years ago
- Training neural networks in TensorFlow 2.0 with 5x less memoryβ134Updated 3 years ago
- Standalone commandline CLI tool for compiling Triton kernelsβ18Updated last year
- Implementation of a Transformer, but completely in Tritonβ274Updated 3 years ago
- Productionize machine learning predictions, with ONNX or withoutβ66Updated last year
- Example code and applications for machine learning on Graphcore IPUsβ330Updated last year
- β21Updated 6 months ago
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodesβ241Updated 2 years ago
- Benchmarks to capture important workloads.β31Updated 7 months ago
- π Interactive performance profiling and debugging tool for PyTorch neural networks.β64Updated 7 months ago
- PyTorch RFCs (experimental)β135Updated 3 months ago
- A Data-Centric Compiler for Machine Learningβ84Updated last year
- A tensor-aware point-to-point communication primitive for machine learningβ271Updated last month
- β252Updated last year
- Fast sparse deep learning on CPUsβ56Updated 2 years ago
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mindβ¦β161Updated 2 months ago
- β330Updated this week