1.0.8 • Published 2 years ago

jupyter-lab-neural-compressor v1.0.8

Weekly downloads
-
License
BSD-3-Clause
Repository
github
Last release
2 years ago

Intel® Neural Compressor as JupyterLab Extension

A JupyterLab Extension library supporting Neural Coder, a novel feature powered by Intel® Neural Compressor providing automatic quantization to further simplify computing performance optimizations of Deep Learning models.

Installation

By Extension Manager in JupyterLab (Recommended)

Search for jupyter-lab-neural-compressor in the Extension Manager in JupyterLab.

By Linux Terminal

npm i jupyter-lab-neural-compressor
jupyter labextension install jupyter-lab-neural-compressor

Getting Started!

As shown in the drop-down list, the supported features include "INT8 (Static Quantization)", "INT8 (Dynamic Quantization)", "BF16", and "Auto Enable & Benchmark". Each of the first three options enables a specific quantization feature into your Deep Learning scripts. The last option automatically enables all quantization features on a Deep Learning script and automatically evaluates the best performance on the model. It is a code-free solution that can help users enable quantization algorithms on a Deep Learning model with no manual coding needed.

Auto-enable a feature

Click the run button on the left side of the drop-down list to start. After finishing, you can see the code changes for the specific optimization enabling as shown in the figure below:

Or let us help you auto-select the best feature

The last option automatically enables each quantization feature on your Deep Learning script and automatically evaluates for the best performance among all features on your Deep Learning model. Since it will automatically run the Python script for benchmark, it requires you to enter additional parameters needed to run your Python script. If there is no additional parameter needed, you can just leave it blank:

Pre-requisites

apt-get update && apt-get install bc numactl
conda install mkl mkl-include jemalloc
pip3 install neural-compressor opencv-python-headless