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
- A tracing JIT compiler for PyTorchβ13Updated 3 years ago
- 3X speedup over Appleβs TensorFlow plugin by using Apache TVM on M1β138Updated 3 years ago
- PyTorch interface for the IPUβ181Updated 2 years ago
- GPU implementation of a fast generalized ANS (asymmetric numeral system) entropy encoder and decoder, with extensions for lossless compreβ¦β356Updated last week
- A thin, highly portable toolkit for efficiently compiling dense loop-based computation.β148Updated 2 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β¦β181Updated 2 months ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.β305Updated last week
- β52Updated last year
- Customized matrix multiplication kernelsβ57Updated 3 years ago
- PyTorch RFCs (experimental)β135Updated 5 months ago
- Fast sparse deep learning on CPUsβ56Updated 3 years ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.β123Updated last month
- An open-source efficient deep learning framework/compiler, written in python.β733Updated 2 months ago
- A tensor-aware point-to-point communication primitive for machine learningβ274Updated 2 months ago
- Home for OctoML PyTorch Profilerβ114Updated 2 years ago
- Example code and applications for machine learning on Graphcore IPUsβ328Updated last year
- Make triton easierβ48Updated last year
- Implementation of a Transformer, but completely in Tritonβ276Updated 3 years ago
- PyTorch centric eager mode debuggerβ48Updated 10 months ago
- Torch Distributed Experimentalβ117Updated last year
- Hacks for PyTorchβ19Updated 2 years ago
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.β45Updated last year
- The Foundation for All Legate Librariesβ231Updated this week
- Training neural networks in TensorFlow 2.0 with 5x less memoryβ136Updated 3 years ago
- Benchmarks to capture important workloads.β31Updated 9 months ago
- β74Updated 2 years ago
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodesβ242Updated 2 years ago
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mindβ¦β161Updated last month
- A library for syntactically rewriting Python programs, pronounced (sinner).β68Updated 3 years ago