Common runtime for machine learning models compiled by AWS SageMaker Neo, TVM, or TreeLite.
Project description
# DLR
DLR is a compact, common runtime for deep learning models and decision tree models compiled by [AWS SageMaker Neo](https://aws.amazon.com/sagemaker/neo/), [TVM](https://tvm.ai/), or [Treelite](https://treelite.readthedocs.io/en/latest/install.html). DLR uses the TVM runtime, Treelite runtime, NVIDIA TensorRT™, and can include other hardware-specific runtimes. DLR provides unified Python/C++ APIs for loading and running compiled models on various devices. DLR currently supports platforms from Intel, NVIDIA, and ARM, with support for Xilinx, Cadence, and Qualcomm coming soon.
## Installation On X86_64 CPU targets running Linux, you can install latest release of DLR package via
pip install dlr
For installation of DLR on GPU targets, non-x86 edge devices, or building DLR from source, please refer to [Installing DLR](https://neo-ai-dlr.readthedocs.io/en/latest/install.html)
## Documentation For instructions on using DLR, please refer to [Amazon SageMaker Neo – Train Your Machine Learning Models Once, Run Them Anywhere](https://aws.amazon.com/blogs/aws/amazon-sagemaker-neo-train-your-machine-learning-models-once-run-them-anywhere/)
Also check out the [API documentation](https://neo-ai-dlr.readthedocs.io/en/latest/)
## Examples We prepared several examples demonstrating how to use DLR API on different platforms
[Neo AI DLR image classification Android example application](examples/android/image_classification)
[DL Model compiler for Android](examples/android/tvm_compiler)
[DL Model compiler for AWS EC2 instances](container/ec2_compilation_container)
## License
This library is licensed under the Apache License Version 2.0.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
Built Distribution
Hashes for dlr-1.1-py2.py3-none-manylinux1_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | c55410d2a56b5b8a50a4dc8c2c7409036c5dd7a83cb2e9e6fef46c64bbebf6e9 |
|
MD5 | 2a348fd86b8c2eeb3318dee8a4fcb275 |
|
BLAKE2b-256 | 0d85fa57902d93b23f7fbbc4d1e2664d67f3b5ebe814232015f095c1b9b8fbd4 |