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 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 8 months ago
- Torch Distributed Experimentalβ117Updated last year
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mindβ¦β161Updated this week
- A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-β¦β67Updated 2 years ago
- Provide Python access to the NVML library for GPU diagnosticsβ246Updated 2 weeks ago
- Home for OctoML PyTorch Profilerβ114Updated 2 years ago
- Lightning HPO & Training Studio Appβ18Updated 2 years ago
- Plugin for deploying MLflow models to TorchServeβ110Updated 2 years ago
- benchmarking some transformer deploymentsβ26Updated 2 years ago
- Accelerate PyTorch models with ONNX Runtimeβ364Updated 7 months ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Moduleβ83Updated last year
- Python 3 Bindings for the NVIDIA Management Libraryβ140Updated 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 3 weeks ago
- Pytorch Lightning Distributed Accelerators using Rayβ213Updated last year
- Reference models for Intel(R) Gaudi(R) AI Acceleratorβ166Updated this week
- TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and supβ¦β389Updated this week
- Productionize machine learning predictions, with ONNX or withoutβ66Updated last year
- EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objaxβ128Updated last year
- Placeholder for the opensource Grid AI componentsβ44Updated 3 years ago
- The Triton backend for the PyTorch TorchScript models.β159Updated 2 weeks ago
- Hacks for PyTorchβ19Updated 2 years ago
- PyTorch RFCs (experimental)β135Updated 3 months ago
- TorchFix - a linter for PyTorch-using code with autofix supportβ148Updated last month
- Automatically insert nvtx ranges to PyTorch modelsβ19Updated 4 years ago
- Unofficial JAX implementations of deep learning research papersβ156Updated 3 years ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.β123Updated last week
- Implementation of a Transformer, but completely in Tritonβ274Updated 3 years ago
- Example code and applications for machine learning on Graphcore IPUsβ330Updated last year
- β109Updated 4 years ago