ov-training-kit 0.1.8

Wrappers for scikit-learn and PyTorch models with OpenVINO optimization

OpenVino training kit

Wrappers for scikit-learn and PyTorch models with OpenVINO optimization.

About

This module provides easy-to-use wrappers for training, evaluating, and exporting classical (scikit-learn) and deep learning (PyTorch) models optimized for OpenVINO, targeting local AI PCs and edge deployment.

System Requirements

• Operating System: Linux (Ubuntu 18.04+), Windows 10/11, Windows Server 2019+
• CPU: x86-64 (Intel or AMD)
• Python: 3.8, 3.9, 3.10, 3.11
• RAM: 8GB+ recommended
• GPU: Optional (not required)
• Note: Intel Extension for PyTorch (IPEX) is only supported on Linux/Windows with x86-64 CPUs. On MacOS, some features may not be available.

Installation

pip install ov-training-kit

Usage

For detailed usage instructions and examples, please refer to the README files inside the src/sklearn and src/pytorch folders.

---

For questions, suggestions, or contributions, feel free to open an issue or pull

🎓 Credits & License

Developed as part of a GSoC

Authors

• Leonardo Heim
• Shivam Basia