Skip to main content

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


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

ai_mk_toolkit-8.8.2.tar.gz (55.9 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

ai_mk_toolkit-8.8.2-py3-none-any.whl (61.0 kB view details)

Uploaded Python 3

File details

Details for the file ai_mk_toolkit-8.8.2.tar.gz.

File metadata

  • Download URL: ai_mk_toolkit-8.8.2.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

Hashes for ai_mk_toolkit-8.8.2.tar.gz
Algorithm Hash digest
SHA256 52796ce65b09f068622f1bc4a381b38666e5bb533f18f2df85263ec854f5ee60
MD5 4eacd9763559d3fb9f1e2866cc2ba5fd
BLAKE2b-256 c5f05368250bc038120c355f9b270e6976dfcad453ba0831ffca5b7c76ae5a82

See more details on using hashes here.

File details

Details for the file ai_mk_toolkit-8.8.2-py3-none-any.whl.

File metadata

  • Download URL: ai_mk_toolkit-8.8.2-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

Hashes for ai_mk_toolkit-8.8.2-py3-none-any.whl
Algorithm Hash digest
SHA256 053712e70df1aced6040bcab9f68579b21139ebe39fc71bf3e15f6161b31d53c
MD5 09a32bb3970261d0787a9b1e04522c45
BLAKE2b-256 745f2641404664be1d37f4b5ed3a139187a336b5c0af927909fbfbe687a6d66c

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page