modelkit 0.0.9

Machine learning lib.

modelkit

Python framework for production ML systems.

---

modelkit is a Python framework meant to make your ML models robust, reusable and performant in all situations you need to use them.

It is meant to bridge the gap between the different uses of your algorithms. With modelkit you can ensure that the same exact code will run in production, on your machine, or on data processing pipelines.

Features

modelkit's key features are:

• simple modelkit is just a Python library, use pip to install it and you are done.
• custom modelkit is useful whenever you need to go beyond off-the-shelf models: custom processing, heuristics, business logic, different frameworks, etc.
• framework agnostic you bring your own framework to the table, and you can use whatever code or library you want. Similarly, modelkit is not opinionated about how you build or train your models.
• organized modelkit encourages you to version and share you ML library and artifacts with others, as a Python package or as a service. Let others use and evaluate your models!
• fast modelkit add minimal overhead to prediction calls. Model predictions can be batched for speed (you define the batching logic).
• fast to code Models only need to define their prediction logic and that's it. No cumbersome pre or postprocessing logic, branching options, etc... The boilerplate code is minimal and sensible.
• fast to deploy Models can be served in a single CLI call using fastapi

And more:

• composable Models can depend on other models, and evaluate them however you need to
• extensible Models can rely on arbitrary supporting configurations files called assets hosted on local or cloud object stores
• type-safe Models' inputs and outputs can be validated by pydantic, you get type annotations for your predictions and can catch errors with static type analysis tools during development.
• async Models support async and sync prediction functions. modelkit supports calling async code from sync code so you don't have to suffer from partially async code.
• testable Models carry their own unit test cases, and unit testing fixtures are available for pytest
• robust modelkit helps you follow software development best practices: all configurations and artifacts are explicitly versioned and tested.

Installation

Install with pip:

pip install modelkit

Documentation

Refer to the documentation for more information.