Model management MLFlow compatibility layer
Everyone can contribute. Help move this issue forward while earning points, leveling up and collecting rewards.
Add a compatibility layer so that users can still rely on their code that uses MLflow to manage their models. To do so, we need to add the following endpoints to https://gitlab.com/gitlab-org/gitlab/-/tree/master/lib/api/ml/mlflow
Registered Models (First Iteration)
- POST registered-models/create (Adding MLFlow API for creating registered models (!134443 - merged))
- GET registered-models/get (Adding MLFlow API for creating registered models (!134443 - merged))
- PATCH registered-models/update (Added MLFLow API to update registered models (!135738 - merged))
- POST registered-models/get-latest-versions (Added MLFLow API to get latest model version (!136117 - merged))
- DELETE registered-models/delete (Added MLFLow API to delete a registered model b... (!136259 - merged))
- GET registered-models/search (Added MLflow API to search for Registered Models (!137476 - merged))
Registered Models (Later Iterations)
- POST registered-models/rename
- POST registered-models/set-tag
- DELETE registered-models/delete-tag
- DELETE registered-models/alias
- POST registered-models/alias
Model Versions
- POST model-versions/create (Adding MLFlow API for creating model versions (!137421 - merged))
- GET model-versions/get (Mlflow endpoint `mlflow/model_versions/get` (!134908 - merged))
- GET model-versions/get-download-uri
- PATCH model-versions/update
- DELETE model-versions/delete
- GET model-versions/search
Note:
POST model-versions/transition-stage is not to be implemented yet, as GitLab Model Registry does not handle release tags yet
Prioritized Roadmap
- CreateModelVersionService should also create pa... (#433103 - closed) (Create Package during Model Version creation (!138230 - merged))
- Add feature toggle for Model registry (#412734 - closed)
- [MLflow Compatibility]: allow users to pass ver... (#434408 - closed) (Added gitlab_tags helper to MLflow API (!139706 - merged))
- [MLflow compatibility] Store ModelVersion tags ... (#434409 - closed) (Adding Model Version Metadata to MLflow (!139975 - merged))
- Allow MarkPackageForDestructionService to be ca... (#432066)
- Resolve Latest version based on SemVer (#428644 - closed) (Adding SemanticVersionable concern to Ml::Model... (!142228 - merged))
- Remaining Model Versions APIs
Future Roadmap Items
Edited by 🤖 GitLab Bot 🤖