project-codeflare / zero-copy-model-loadingLinks
In-depth code associated with my Medium blog post, "How to Load PyTorch Models 340 Times Faster with Ray"
☆28Updated 3 years ago
Alternatives and similar repositories for zero-copy-model-loading
Users that are interested in zero-copy-model-loading are comparing it to the libraries listed below
Sorting:
- FIL backend for the Triton Inference Server☆87Updated last week
- Provide Python access to the NVML library for GPU diagnostics☆258Updated 5 months ago
- TorchFix - a linter for PyTorch-using code with autofix support☆152Updated 5 months ago
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆164Updated 3 weeks ago
- Module, Model, and Tensor Serialization/Deserialization☆286Updated 5 months ago
- TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and sup…☆412Updated this week
- 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
- Unified storage framework for the entire machine learning lifecycle☆155Updated last year
- The Triton backend for the PyTorch TorchScript models.☆173Updated this week
- Simple dependency injection framework for Python☆21Updated last year
- The Triton backend for the ONNX Runtime.☆172Updated this week
- 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
- Triton Model Navigator is an inference toolkit designed for optimizing and deploying Deep Learning models with a focus on NVIDIA GPUs.☆216Updated last week
- 🏙 Interactive performance profiling and debugging tool for PyTorch neural networks.☆64Updated last year
- Productionize machine learning predictions, with ONNX or without☆66Updated 2 years ago
- Distributed XGBoost on Ray☆152Updated last year
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆328Updated last week
- Dynamic batching library for Deep Learning inference. Tutorials for LLM, GPT scenarios.☆106Updated last year
- Python 3 Bindings for the NVIDIA Management Library☆142Updated last year
- Torch Distributed Experimental☆117Updated last year
- Pytorch Lightning Distributed Accelerators using Ray☆215Updated 2 years ago
- Python client for RedisAI☆89Updated 2 years ago
- Home for OctoML PyTorch Profiler☆113Updated 2 years ago
- Plugin for deploying MLflow models to TorchServe☆110Updated 2 years ago
- A memory efficient DLRM training solution using ColossalAI☆105Updated 3 years ago
- 🐍 Python bidding for the Hora Approximate Nearest Neighbor Search Algorithm library☆73Updated 4 years ago
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆46Updated last year
- 🕹️ Performance Comparison of MLOps Engines, Frameworks, and Languages on Mainstream AI Models.☆138Updated last year
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆130Updated last week
- Easy and lightning fast training of 🤗 Transformers on Habana Gaudi processor (HPU)☆205Updated this week