Model management MLFlow compatibility layer
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) (Adding SemanticVersionable concern to Ml::Model... (!142228 - merged)) -
Remaining Model Versions APIs
Future Roadmap Items
Edited by Darby Frey