NodLabs / shark-samples
☆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 …☆106Updated 4 months ago
- Documentation autogeneration utilities.☆30Updated 2 years ago
- Personal collection of references for high performance mixed precision training.☆41Updated 5 years ago
- benchmarking some transformer deployments☆26Updated 2 years ago
- A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…☆67Updated last year
- This repository contains example code to build models on TPUs☆30Updated 2 years ago
- Home for OctoML PyTorch Profiler☆113Updated 2 years ago
- Torch Distributed Experimental☆115Updated 9 months ago
- 3X speedup over Apple’s TensorFlow plugin by using Apache TVM on M1☆136Updated 3 years ago
- Train ImageNet in 18 minutes on AWS☆129Updated last year
- PyProf2: PyTorch Profiling tool☆82Updated 4 years ago
- Lightning HPO & Training Studio App☆18Updated 2 years ago
- EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax☆128Updated last year
- Pygloo provides Python bindings for Gloo.☆22Updated 2 months ago
- Notes and artifacts from the ONNX steering committee☆26Updated this week
- Minimal implementation of adaptive gradient clipping (https://arxiv.org/abs/2102.06171) in TensorFlow 2.☆84Updated 3 years ago
- Productionize machine learning predictions, with ONNX or without☆65Updated last year
- Code for scaling Transformers☆26Updated 4 years ago
- pytest plugin for a better developer experience when working with the PyTorch test suite☆44Updated 3 years ago
- Implémentation of the article **Deep Learning CUDA Memory Usage and Pytorch optimization tricks**☆43Updated 5 years ago
- Lightweight interface to AWS☆47Updated 5 years ago
- Accelerate PyTorch models with ONNX Runtime☆360Updated 2 months ago
- Simple tooling for marking deprecated functions or classes and re-routing to the new successors' instance.☆51Updated last month
- A collection of Models, Datasets, DataModules, Callbacks, Metrics, Losses and Loggers to better integrate pytorch-lightning with transfor…☆47Updated last year
- PyTorch interface for the IPU☆179Updated last year
- Reference implementations of popular Binarized Neural Networks☆107Updated last month
- ☆38Updated 2 years ago
- A case study of efficient training of large language models using commodity hardware.☆69Updated 2 years ago
- Document parameters using comments☆10Updated 3 years ago
- Various transformers for FSDP research☆37Updated 2 years ago