Skip to main content

Distributed Scale Out Framework for ML models serving/inference

Project description


Distributed Scale Out Framework for ML model serving/inferencing

This Project is in development, Not intended for production use


It is available in PyPI

$ pip install inferout

if you don't have pip command use -m pip

$ python -m pip install inferout

Usage (Quick example)

  • First of all we need an ml model to be able to serve. It can be of any kind, any framework (pytorch, tensorflow, rasa, etc.) as long as it can be loaded in python 3.7+

  • We need to implement two interfaces

    • serving_engine - this is to teach inferout how to load, serve (infer), and unload the models of a specific kind/usecase. example
    • storage_engine - this is to teach inferout how to get/download/locate the models. example
  • Get redis server ready. Yes we need redis to use inferout. inferout uses redis to store metadata and pass messages between diferent componets and nodes.

  • Bootstrap cluster - this is create minimum metadata required to run cluster all you need is a cluster name and a redis URL

    $ inferout bootstrap_cluster --cluster-name my_rasa_nlu --redis-url redis:///10
  • Launch the worker

    $ inferout worker --cluster-name my_rasa_nlu --redis-url redis:///10 --storage-engines "my_rasa_nlu.storage_engine" --serving-engines "my_rasa_nlu.serving_engine"

    Can run multiple workers for single cluster. to run multiple workers in single system (for development and to test) we can use diferent port numbers for each worker. try "inferout worker -h" for more details

    What we need to make sure is worker availabily and connectivity(serving api port) between nodes else(replication of models and distributing to workers, smartly routing the inferencing requests) will be taken care by inferout framework.

  • Create namespace

    $ curl -XPUT localhost:9500/namespaces/default -d '{"settings":{"storage_engine":"my_rasa_nlu.storage_engine","serving_engine":"my_rasa_nlu.serving_engine"}}'
  • Create model

    $ curl -XPUT localhost:9500/namespaces/default/models/mymodel1 -d '{"parameters":{"path":"nlu-20210726-153112.tar.gz"}}'

    wondering how to get this model file and where to place it for testing? install open source rasa using pip.

    $ pip3 install rasa

    init rasa project

    $ rasa init

    train your nlu model

    $ rasa train nlu

    make required directories and copy the model

    $ mkdir /tmp/nlu_models
    $ mkdir /temp_nlu_models
    $ ls models
    $ cp models/nlu-*.tar.gz /tmp/nlu_models/
  • Query your model (inference)

    $ curl -XPOST localhost:9510/default/mymodel1 -d '{"input_data":{"query":"Hi"}}'

    Did you find any change in port number? Yes for namespace and model creation we used 9500 but now we used 9510 inferout worker provides 2 API services

    • management API - create/update/delete/inspect models, namespaces, workers
    • inferencing API - to quiry models
  • What next? Explore other management APIs, for now just find API endpoints in source code

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

inferout-0.0.0a5.tar.gz (17.1 kB view hashes)

Uploaded Source

Supported by

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