Unified AI/ML package providing a consistent API for accessing 205 models and tasks from scikit-learn, PyTorch, TensorFlow, and other popular libraries.
Project description
AI_MK_Toolkit
AI_MK_Toolkit is a production-ready Python package that provides a unified, consistent interface for accessing over 205 machine learning, deep learning, natural language processing (NLP), computer vision (CV), and reinforcement learning (RL) models and tasks from leading open-source frameworks.
Rather than reimplementing algorithms, AI_MK_Toolkit acts as a lightweight abstraction layer that simplifies the use of multiple AI frameworks while preserving the functionality, performance, and flexibility of their official implementations.
Python Requirement: Python 3.10 or later
Supported Platforms: Windows, Linux, and macOS
๐ Why AI_MK_Toolkit?
Modern AI development often requires switching between multiple frameworks, each with its own APIs, import patterns, dependency management, and usage conventions. AI_MK_Toolkit removes this complexity by providing a single, intuitive, and consistent interface across popular AI ecosystems, allowing developers to focus on building solutions instead of framework-specific boilerplate.
โจ Key Features
- Unified API โ Access models and tasks from multiple AI frameworks through a consistent interface.
- 205+ Supported Entries โ Includes 51 Machine Learning models, 67 Deep Learning models, 54 NLP models, 19 Reinforcement Learning algorithms, and 14 Computer Vision task APIs.
- Lazy Loading โ Optional dependencies are imported only when required, reducing startup time and installation overhead.
- Framework Agnostic โ Use models from libraries such as scikit-learn, PyTorch, TensorFlow, XGBoost, LightGBM, Hugging Face Transformers, Stable-Baselines3, and more through a common API.
- Production Ready โ Built with type hints, comprehensive error handling, and PEP 561 (
py.typed) support. - Modular Installation โ Install only the dependencies needed for your specific use case using optional extras.
- Transparent Attribution โ Clearly documents the original projects and their licenses, ensuring proper acknowledgment of all underlying open-source libraries.
- Developer Friendly โ Simplifies experimentation, prototyping, education, and production workflows by minimizing framework-specific code.
๐ About This Project
AI-MK-Toolkit is a production-ready Python package that provides a unified, consistent interface to over 205 machine learning, deep learning, natural language processing, computer vision, reinforcement learning, and large language model (LLM) implementations from widely adopted open-source frameworks.
Rather than reimplementing algorithms, AI-MK-Toolkit serves as a lightweight abstraction layer that enables developers to access models from multiple ecosystems through a single, intuitive API while preserving the functionality and performance of the underlying libraries.
Key Principles
- Unified API โ Access models from multiple frameworks using a consistent interface.
- No Algorithm Reimplementation โ Uses the official implementations provided by libraries such as scikit-learn, PyTorch, TensorFlow, XGBoost, LightGBM, Hugging Face Transformers, Stable-Baselines3, and others.
- Modular Installation โ Install only the dependencies required for your specific use case through optional extras.
- Transparent Attribution โ Properly acknowledges and respects the licenses of all underlying open-source projects.
- Developer-Friendly โ Simplifies experimentation, prototyping, and production development by reducing framework-specific boilerplate.
Important: AI-MK-Toolkit is an interface and integration library. It does not claim ownership of, modify, or replace the original machine learning algorithms or pretrained models provided by the respective open-source projects. All credit for the underlying implementations belongs to their original authors and maintainers.
๐ฆ Installation
Base Installation
pip install ai-mk-toolkit
Deep Learning
pip install ai-mk-toolkit[dl]
Natural Language Processing
pip install ai-mk-toolkit[nlp]
Reinforcement Learning
pip install ai-mk-toolkit[rl]
Complete Installation
pip install ai-mk-toolkit[full]
๐ Quick Start
Machine Learning
from ai_mk_toolkit import ml
model = ml.RandomForest()
model.fit(X_train, y_train)
predictions = model.predict(X_test)
Deep Learning
from ai_mk_toolkit import dl
model = dl.ResNet50(pretrained=True)
Natural Language Processing
from ai_mk_toolkit import nlp
bert = nlp.BERT()
tokenizer = nlp.Tokenizer()
Reinforcement Learning
from ai_mk_toolkit import rl
agent = rl.DQN(policy="MlpPolicy", env=env)
Computer Vision
from ai_mk_toolkit import cv
detector = cv.ObjectDetection()
classifier = cv.ImageClassification()
๐ Package Overview
| Module | Description |
|---|---|
| ML | 51 classical machine learning models |
| DL | 67 deep learning model wrappers |
| NLP | 54 NLP models and task APIs |
| RL | 19 reinforcement learning algorithms |
| CV | 14 computer vision task APIs |
Total: 205+ unified model wrappers and task interfaces
๐ Package Structure
ai_mk_toolkit/
โโโ ml/
โโโ dl/
โโโ nlp/
โโโ rl/
โโโ cv/
โโโ automl/
โโโ datasets/
โโโ preprocessing/
โโโ metrics/
โโโ explainability/
โโโ visualization/
โโโ deployment/
โโโ plugins/
โโโ backends/
โโโ core/
โ Quality
- Production-ready package structure
- Type annotated (PEP 561)
- Comprehensive automated test suite
- Wheel and source distributions
- Lazy-loading architecture
- Clean wrapper design (no algorithm reimplementation)
๐งช Testing
pytest
pytest -v
๐ License
AI_MK_Toolkit is distributed under the MIT License.
The package wraps existing open-source libraries, each of which retains its own license.
Important: Some optional dependencies have copyleft licenses (for example, Ultralytics (AGPL-3.0) and Gensim (LGPL-2.1+)). Users are responsible for complying with the licenses of any optional dependencies they choose to install.
๐ Acknowledgments
AI_MK_Toolkit builds upon the excellent work of the open-source community.
Major frameworks include:
- scikit-learn
- NumPy
- PyTorch
- torchvision
- TensorFlow / Keras
- Hugging Face Transformers
- XGBoost
- LightGBM
- CatBoost
- Stable-Baselines3
- timm
- spaCy
- NLTK
- Gensim
- sentence-transformers
- Ultralytics
Important: All trademarks, models, algorithms, and implementations remain the intellectual property of their respective authors and maintainers.
๐ค Contributing
Contributions, feature requests, and bug reports are welcome.
Before contributing:
- Follow the existing project structure.
- Document any new dependency and its license.
- Prefer permissive licenses (MIT, BSD, Apache-2.0).
- Keep optional dependencies modular whenever possible.
โ Disclaimer
AI_MK_Toolkit is a wrapper library, not a machine learning framework.
Important: It provides a unified Python interface to official implementations from existing open-source projects and does not claim ownership or authorship of the underlying algorithms or pretrained models. All credit belongs to the original authors and maintainers of those projects.
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 Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file ai_mk_toolkit-8.8.0.tar.gz.
File metadata
- Download URL: ai_mk_toolkit-8.8.0.tar.gz
- Upload date:
- Size: 55.9 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.10.11
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
ee4026fd1e6ebb86fa0e4caf08dd83273f4c3e5197ed1bdb36ec2b8c8c2de962
|
|
| MD5 |
a6b8fbc01eb8c42c991880b9092e0dde
|
|
| BLAKE2b-256 |
593988430d2b52204dbf0d578b9c5c459e8a925fb3e7674d6059671caefe4875
|
File details
Details for the file ai_mk_toolkit-8.8.0-py3-none-any.whl.
File metadata
- Download URL: ai_mk_toolkit-8.8.0-py3-none-any.whl
- Upload date:
- Size: 61.0 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.10.11
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
e474f672ea40c2e21f0c87712d7e49dacc2437cb6566ab81f384edffce4043d2
|
|
| MD5 |
91ec4f2682169f199853592cfb49965e
|
|
| BLAKE2b-256 |
891afd04cf0001623ba756665e63e29db0b218706042bd23353594733687c71a
|