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:
- PyTorch interface for the IPU☆181Updated 2 years ago
- Python 3 Bindings for the NVIDIA Management Library☆141Updated last year
- Torch Distributed Experimental☆117Updated last year
- 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 9 months ago
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆161Updated 3 weeks ago
- Provide Python access to the NVML library for GPU diagnostics☆248Updated last month
- EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax☆128Updated last year
- Pytorch Lightning Distributed Accelerators using Ray☆214Updated last year
- Reference models for Intel(R) Gaudi(R) AI Accelerator☆165Updated 3 weeks ago
- Placeholder for the opensource Grid AI components☆44Updated 3 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…☆180Updated last month
- TorchFix - a linter for PyTorch-using code with autofix support☆149Updated last month
- Lightning HPO & Training Studio App☆18Updated 2 years ago
- 3X speedup over Apple’s TensorFlow plugin by using Apache TVM on M1☆138Updated 3 years ago
- Using the famous cnn model in Pytorch, we run benchmarks on various gpu.☆245Updated last year
- PyTorch RFCs (experimental)☆135Updated 4 months ago
- Home for OctoML PyTorch Profiler☆114Updated 2 years ago
- This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic …☆102Updated this week
- PyTorch centric eager mode debugger☆48Updated 10 months ago
- A python library that provides common I/O interface across different storage backends.☆147Updated 3 weeks ago
- A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…☆66Updated 2 years ago
- Customized matrix multiplication kernels☆57Updated 3 years ago
- Prototype routines for GPU quantization written using PyTorch.☆21Updated 2 months ago
- A tensor-aware point-to-point communication primitive for machine learning☆273Updated 2 months ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆122Updated last month
- A logging tool for deep learning.☆60Updated 6 months ago
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆251Updated 2 years ago
- Easily benchmark PyTorch model FLOPs, latency, throughput, allocated gpu memory and energy consumption☆107Updated 2 years ago
- Implementation of a Transformer, but completely in Triton☆275Updated 3 years ago
- Continuous builder and binary build scripts for pytorch☆354Updated 2 months ago