MLflow
MLflow provides experiment tracking for ULTRATHINK research projects on the MEGATRON cluster.
Overview
| Property | Value |
|---|---|
| Namespace | ai-ml |
| Access | Internal (ClusterIP) |
| Purpose | ML experiment tracking, model registry, artifact storage |
Usage with ULTRATHINK
Each research project logs experiments to MLflow:
import mlflow
mlflow.set_tracking_uri("http://mlflow.ai-ml.svc.cluster.local:5000")
mlflow.set_experiment("rosa-vit")
with mlflow.start_run():
mlflow.log_param("learning_rate", 0.001)
mlflow.log_metric("accuracy", 0.95)
mlflow.log_artifact("model.pt")
kubectl get pods -n ai-ml -l app=mlflow
TODO: expand with experiment inventory, model registry contents, and artifact storage configuration.