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:
- Simple dependency injection framework for Python☆21Updated last year
- Module, Model, and Tensor Serialization/Deserialization☆283Updated 4 months ago
- Provide Python access to the NVML library for GPU diagnostics☆257Updated 4 months ago
- TorchFix - a linter for PyTorch-using code with autofix support☆152Updated 4 months ago
- WIP. Veloce is a low-code Ray-based parallelization library that makes machine learning computation novel, efficient, and heterogeneous.☆17Updated 3 years ago
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆162Updated 3 weeks ago
- Python client for RedisAI☆89Updated 2 years ago
- Productionize machine learning predictions, with ONNX or without☆66Updated 2 years ago
- Python 3 Bindings for the NVIDIA Management Library☆142Updated last year
- benchmarking some transformer deployments☆26Updated last month
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆47Updated 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 weeks ago
- A library that translates Python and NumPy to optimized distributed systems code.☆131Updated 3 years ago
- The Triton backend for the PyTorch TorchScript models.☆170Updated this week
- Distributed XGBoost on Ray☆152Updated last year
- A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…☆67Updated 2 years ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆325Updated last week
- FIL backend for the Triton Inference Server☆87Updated last month
- 🐍 Python bidding for the Hora Approximate Nearest Neighbor Search Algorithm library☆73Updated 4 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 last month
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆46Updated last year
- Home for OctoML PyTorch Profiler☆114Updated 2 years ago
- Distributed ML Optimizer☆34Updated 4 years ago
- TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and sup…☆411Updated last week
- High-performance safetensors model loader☆92Updated 3 weeks ago
- Ray - A curated list of resources: https://github.com/ray-project/ray☆78Updated 2 months ago
- A collection of reproducible inference engine benchmarks☆38Updated 8 months ago
- Distributed skorch on Ray Train☆58Updated 3 years ago
- ML/DL Math and Method notes☆66Updated 2 years ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆130Updated 3 weeks ago