NodLabs / shark-samplesLinks
β15Updated 3 years ago
Alternatives and similar repositories for shark-samples
Users that are interested in shark-samples are comparing it to the libraries listed below
Sorting:
- Nod.ai π¦ version of π» . You probably want to start at https://github.com/nod-ai/shark for the product and the upstream IREE repository β¦β107Updated last month
- PyTorch interface for the IPUβ181Updated 2 years ago
- A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-β¦β67Updated 2 years ago
- Torch Distributed Experimentalβ117Updated last year
- Python 3 Bindings for the NVIDIA Management Libraryβ142Updated last year
- Pytorch Lightning Distributed Accelerators using Rayβ215Updated 2 years ago
- EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objaxβ129Updated 2 years ago
- Accelerate PyTorch models with ONNX Runtimeβ367Updated last week
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mindβ¦β164Updated 3 weeks 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β¦β182Updated last month
- Example code and applications for machine learning on Graphcore IPUsβ333Updated last year
- Home for OctoML PyTorch Profilerβ113Updated 2 years ago
- Productionize machine learning predictions, with ONNX or withoutβ66Updated 2 years ago
- TorchFix - a linter for PyTorch-using code with autofix supportβ152Updated 5 months ago
- Placeholder for the opensource Grid AI componentsβ45Updated 3 years ago
- pytest plugin for a better developer experience when working with the PyTorch test suiteβ44Updated 4 years ago
- Prototype routines for GPU quantization written using PyTorch.β21Updated 3 weeks ago
- Training material for IPU users: tutorials, feature examples, simple applicationsβ87Updated 2 years ago
- The Triton backend for the PyTorch TorchScript models.β172Updated 2 weeks ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.β327Updated 3 weeks ago
- Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorchβ184Updated 3 years ago
- Implementation of a Transformer, but completely in Tritonβ279Updated 3 years ago
- Lightning HPO & Training Studio Appβ19Updated 2 years ago
- PyTorch RFCs (experimental)β138Updated 8 months ago
- 3X speedup over Appleβs TensorFlow plugin by using Apache TVM on M1β137Updated 3 years ago
- Python client for RedisAIβ89Updated 2 years ago
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.β250Updated 3 years ago
- Benchmarks to capture important workloads.β32Updated last week
- Performance of PyTorch on Apple Siliconβ51Updated 2 years ago
- Using the famous cnn model in Pytorch, we run benchmarks on various gpu.β247Updated last year