This repo contains example code for a (very basic) ML platform.
- The model-template directory contains an example for a Cookiecutter-based template that data scientists can clone to start a new project.
- The infra directory contains Pulumi code that spins up the shared infrastructure of the ML platform, such as Kubernetes, MLFlow, etc.
As data science teams become more mature with models reaching actual production, the need for a proper infrastructure becomes crucial. Leading companies in the field with massive engineering teams like Uber, Netflix and Airbnb had created multiple solutions for their infrastructure and named the combination of them as “ML Platform”.
We hope this repo can help you get started with building your own ML platform ❤️
- FastAPI - for model serving
- MLFlow - for experiment tracking
- DVC - for data versioning
- Cookiecutter - for the model template
- Pulumi - Infrastructure as Code
- GitHub Actions - for CI/CD
- Traefik - API gateway
- Poetry - Python dependency management
When building your own ML platform, do not take these tools for granted! Check out alternatives and find the best tools that solve each one of your problems.
Well... a lot actually. Here's a partial list:
- HTTPS & Authentication
- Environments (staging, production)
- Common library for preprocessing, postprocessing, etc
- Model input & validation
- Training orchestration
- and probably much more!
We would love your help!