Issue 131: MLflow integration#132

Merged

davramov merged 8 commits intoals-computing:mainfrom

davramov:mlflow-integration

Apr 23, 2026

Commits on Apr 6, 2026

Adding mlflow uris to config
davramov
committed
updating dependencies to include mlflow
davramov
committed
Adding utility script for interacting with mlflow servers (register a model checkpoint, see if server is reachable, get client, retrieve checkpoint info, log run metrics)
davramov
committed
Added a script to bl832 for registering model checkpoints to mlflow (including inference parameter settings). Also includes a query to check that values are pulled from the server as expected.
davramov
committed
Updating _load_job_options() in nersc.py to additionally prioritize model information found in mlflow
davramov
committed
adding pytest for mlflow
davramov
committed
adjusting _load_job_options() in segmentation_sam3() and segmentation_dinov3()
davramov
committed
Swapping sync globus transfers to async for nersc recon/segment flows
davramov
committed