Awesome Infra for AI › Model Serving Frameworks

eightBEC/fastapi-ml-skeleton

⭐ 601 Python repository created 2020-02-15

This project provides a robust boilerplate for serving machine learning models using FastAPI, focusing on production readiness. It offers a structured foundation that accelerates the development and deployment of ML inference endpoints. The skeleton includes pre-configured tools for linting, formatting, static analysis (isort, mypy, flake, black, bandit), and testing, ensuring high code quality from the outset. It supports Python 3.11+ and uses Poetry for dependency management. The project demonstrates how to set up an API with authentication using API keys and includes a sample regression model for house price prediction to illustrate usage. This setup allows developers to quickly integrate their trained ML models, expose them as RESTful APIs, and manage security through API keys. It emphasizes ease of use, fast performance, and a secure environment for deploying machine learning models into production.

https://github.com/eightBEC/fastapi-ml-skeleton

fastapimachine-learningmodel-servingpythonpython3production-readyapiinference

Also in Model Serving Frameworks

vllm-project/vllm

vLLM is a high-throughput and memory-efficient serving and inference engine for large language models, featuring PagedAttention, continuous batching, and extensive hardware and model support.

jundot/omlx

oMLX is an LLM inference server optimized for Apple Silicon, offering continuous batching, tiered KV caching, and a macOS menu bar interface for managing models locally.

jina-ai/serve

A framework for building and deploying cloud-native AI services, with native support for ML frameworks, high-performance serving, LLM streaming, and Kubernetes/Docker Compose deployment.

bentoml/OpenLLM

OpenLLM allows developers to self-host and run any open-source or custom LLMs as OpenAI-compatible API endpoints in the cloud, streamlining deployment and serving.

bentoml/BentoML

BentoML is an open-source framework for building, shipping, and scaling AI applications, providing tools to serve AI/ML models as production-ready API endpoints.

vllm-project/vllm-omni

vLLM-Omni is an extension of vLLM designed for efficient serving and inference of omni-modality AI models, encompassing text, image, video, and audio data processing.

kserve/kserve

KServe is a standardized, distributed platform for serving generative and predictive AI models on Kubernetes, providing scalable deployment and management.

Michael-A-Kuykendall/shimmy

Shimmy is a pure-Rust WebGPU inference engine providing OpenAI-compatible endpoints for local GGUF models, featuring Airframe engine and TurboShimmy INT4 KV cache compression for efficient GPU util...