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
- 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 this week
- Torch Distributed Experimentalβ117Updated last year
- Pytorch Lightning Distributed Accelerators using Rayβ215Updated 2 years ago
- pytest plugin for a better developer experience when working with the PyTorch test suiteβ44Updated 3 years ago
- Python 3 Bindings for the NVIDIA Management Libraryβ141Updated 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β¦β182Updated 3 months ago
- benchmarking some transformer deploymentsβ26Updated this week
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mindβ¦β161Updated 2 months ago
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.β251Updated 3 years ago
- EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objaxβ129Updated last year
- Accelerate PyTorch models with ONNX Runtimeβ367Updated 9 months ago
- 3X speedup over Appleβs TensorFlow plugin by using Apache TVM on M1β138Updated 3 years ago
- Placeholder for the opensource Grid AI componentsβ45Updated 3 years ago
- β108Updated 4 years ago
- Example code and applications for machine learning on Graphcore IPUsβ332Updated last year
- Home for OctoML PyTorch Profilerβ114Updated 2 years ago
- Lightning HPO & Training Studio Appβ18Updated 2 years ago
- Recipes are a standard, well supported set of blueprints for machine learning engineers to rapidly train models using the latest researchβ¦β334Updated this week
- Provide Python access to the NVML library for GPU diagnosticsβ251Updated 2 months ago
- A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-β¦β67Updated 2 years ago
- Customized matrix multiplication kernelsβ57Updated 3 years ago
- PyTorch RFCs (experimental)β136Updated 6 months ago
- Train ImageNet in 18 minutes on AWSβ134Updated last year
- TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and supβ¦β402Updated this week
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodesβ242Updated 2 years ago
- Plugin for deploying MLflow models to TorchServeβ110Updated 2 years ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.β125Updated 2 months ago
- DiffQ performs differentiable quantization using pseudo quantization noise. It can automatically tune the number of bits used per weight β¦β236Updated 2 years ago
- TorchFix - a linter for PyTorch-using code with autofix supportβ151Updated 3 months ago