Catalog PyPI
Back to packageRELEASE DETAILS

8.8.8

PyPI

ai-mk-toolkit

Unified AI/ML package providing a consistent API for accessing 205 models and tasks from scikit-learn, PyTorch, TensorFlow, and other popular libraries.

aicomputer-visiondeep-learningmachine-learningnlpreinforcement-learningunified-api
Published19 Jul 2026
Direct dependencies26
Transitive dependenciesNot calculated

README

Version 8.8.8 documentation

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

ModuleDescription
ML51 classical machine learning models
DL67 deep learning model wrappers
NLP54 NLP models and task APIs
RL19 reinforcement learning algorithms
CV14 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.

Downloads & popularity

Awaiting observations

No metric observations yet

Download counts and popularity will appear here as the collector receives them.

Declared dependencies 44

Native requirements
26 direct targetsNot calculated transitive count
Package / targetRequirementRelationship
numpy
Environment
{
  "extras": "",
  "marker": null,
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
numpy<3,>=1.24runtime
scikit-learn
Environment
{
  "extras": "",
  "marker": null,
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
scikit-learn<2,>=1.3runtime
pyyaml
Environment
{
  "extras": "",
  "marker": null,
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
pyyaml<7,>=6.0runtime
joblib
Environment
{
  "extras": "",
  "marker": null,
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
joblib<2,>=1.3runtime
torch
Environment
{
  "extras": "",
  "marker": null,
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
torch<3,>=2.2runtime
torchvision
Environment
{
  "extras": "",
  "marker": null,
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
torchvision<1,>=0.17runtime
tensorflow
Environment
{
  "extras": "",
  "marker": null,
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
tensorflow<3,>=2.16runtime
ultralytics
Environment
{
  "extras": "",
  "marker": null,
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
ultralytics<9,>=8.3runtime
xgboost
Environment
{
  "extras": "",
  "marker": "extra == \"ml-boosting\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
xgboost<3,>=2.1; extra == "ml-boosting"runtime
lightgbm
Environment
{
  "extras": "",
  "marker": "extra == \"ml-boosting\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
lightgbm<5,>=4.5; extra == "ml-boosting"runtime
catboost
Environment
{
  "extras": "",
  "marker": "extra == \"ml-boosting\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
catboost<2,>=1.2; extra == "ml-boosting"runtime
umap-learn
Environment
{
  "extras": "",
  "marker": "extra == \"ml-boosting\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
umap-learn<1,>=0.5; extra == "ml-boosting"runtime
timm
Environment
{
  "extras": "",
  "marker": "extra == \"dl\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
timm<2,>=1.0; extra == "dl"runtime
ultralytics
Environment
{
  "extras": "",
  "marker": "extra == \"dl\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
ultralytics<9,>=8.3; extra == "dl"runtime
nltk
Environment
{
  "extras": "",
  "marker": "extra == \"nlp\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
nltk<4,>=3.9; extra == "nlp"runtime
spacy
Environment
{
  "extras": "",
  "marker": "extra == \"nlp\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
spacy<4,>=3.7; extra == "nlp"runtime
gensim
Environment
{
  "extras": "",
  "marker": "extra == \"nlp\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
gensim<5,>=4.3; extra == "nlp"runtime
transformers
Environment
{
  "extras": "",
  "marker": "extra == \"nlp\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
transformers<5,>=4.45; extra == "nlp"runtime
sentence-transformers
Environment
{
  "extras": "",
  "marker": "extra == \"nlp\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
sentence-transformers<4,>=3.0; extra == "nlp"runtime
stable-baselines3
Environment
{
  "extras": "",
  "marker": "extra == \"rl\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
stable-baselines3<3,>=2.3; extra == "rl"runtime
sb3-contrib
Environment
{
  "extras": "",
  "marker": "extra == \"rl\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
sb3-contrib<3,>=2.3; extra == "rl"runtime
ultralytics
Environment
{
  "extras": "",
  "marker": "extra == \"cv\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
ultralytics<9,>=8.3; extra == "cv"runtime
transformers
Environment
{
  "extras": "",
  "marker": "extra == \"cv\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
transformers<5,>=4.45; extra == "cv"runtime
matplotlib
Environment
{
  "extras": "",
  "marker": "extra == \"viz\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
matplotlib<4,>=3.8; extra == "viz"runtime
seaborn
Environment
{
  "extras": "",
  "marker": "extra == \"viz\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
seaborn<1,>=0.13; extra == "viz"runtime
xgboost
Environment
{
  "extras": "",
  "marker": "extra == \"full\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
xgboost<3,>=2.1; extra == "full"runtime
lightgbm
Environment
{
  "extras": "",
  "marker": "extra == \"full\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
lightgbm<5,>=4.5; extra == "full"runtime
catboost
Environment
{
  "extras": "",
  "marker": "extra == \"full\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
catboost<2,>=1.2; extra == "full"runtime
umap-learn
Environment
{
  "extras": "",
  "marker": "extra == \"full\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
umap-learn<1,>=0.5; extra == "full"runtime
timm
Environment
{
  "extras": "",
  "marker": "extra == \"full\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
timm<2,>=1.0; extra == "full"runtime
ultralytics
Environment
{
  "extras": "",
  "marker": "extra == \"full\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
ultralytics<9,>=8.3; extra == "full"runtime
nltk
Environment
{
  "extras": "",
  "marker": "extra == \"full\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
nltk<4,>=3.9; extra == "full"runtime
spacy
Environment
{
  "extras": "",
  "marker": "extra == \"full\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
spacy<4,>=3.7; extra == "full"runtime
gensim
Environment
{
  "extras": "",
  "marker": "extra == \"full\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
gensim<5,>=4.3; extra == "full"runtime
transformers
Environment
{
  "extras": "",
  "marker": "extra == \"full\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
transformers<5,>=4.45; extra == "full"runtime
sentence-transformers
Environment
{
  "extras": "",
  "marker": "extra == \"full\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
sentence-transformers<4,>=3.0; extra == "full"runtime
stable-baselines3
Environment
{
  "extras": "",
  "marker": "extra == \"full\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
stable-baselines3<3,>=2.3; extra == "full"runtime
sb3-contrib
Environment
{
  "extras": "",
  "marker": "extra == \"full\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
sb3-contrib<3,>=2.3; extra == "full"runtime
matplotlib
Environment
{
  "extras": "",
  "marker": "extra == \"full\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
matplotlib<4,>=3.8; extra == "full"runtime
seaborn
Environment
{
  "extras": "",
  "marker": "extra == \"full\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
seaborn<1,>=0.13; extra == "full"runtime
build
Environment
{
  "extras": "",
  "marker": "extra == \"dev\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
build<2,>=1.2; extra == "dev"runtime
pytest
Environment
{
  "extras": "",
  "marker": "extra == \"dev\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
pytest<9,>=8.2; extra == "dev"runtime
ruff
Environment
{
  "extras": "",
  "marker": "extra == \"dev\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
ruff<1,>=0.5; extra == "dev"runtime
mypy
Environment
{
  "extras": "",
  "marker": "extra == \"dev\"",
  "requires_python": ">=3.10",
  "evaluation": "NOT_EVALUATED"
}
mypy<2,>=1.10; extra == "dev"runtime

Distribution files

2 collected
ai_mk_toolkit-8.8.8-py3-none-any.whl

bdist_wheel · 59.84 KB · Python py3

Source file
ai_mk_toolkit-8.8.8.tar.gz

sdist · 54.79 KB · Python source

Source file

Metadata history

Meaningful changes
Current collected state Current

Observed since 21 Oct, 11:19 UTC

No superseded metadata states collected. Unchanged observations do not create new revisions.

Record identity

package_version:448ea5e30d42447799a6e025885bd889e5fa163e51d86ab8187b84c41e48ecff

Other collected versions

View all collected versions →