Environment and reproducibility

mlenv

Snapshot the ML stack to a file and diff two snapshots.

Install

pip install mlenv-cli

Once published to PyPI. Also available now from GitHub:

pip install git+https://github.com/jmweb-org/mlenv

What it does

The classic "it trained on my machine" usually comes down to something that moved: the CUDA build behind PyTorch, a minor Python bump, a driver. mlenv captures all of it in one JSON you can commit and compare.

Features

—Python, platform, packages, CUDA, cuDNN and GPUs.
—Diff with the dangerous changes ranked first.
—Optional GPU capture through NVML.
—CI gate and JSON output.

View the code on GitHub →

Other tools

dsdiff

A git-style diff between two datasets, with distribution drift.

splitcheck

Detect rows leaking between train, validation and test splits.

pii-sweep

Scan datasets for PII before sharing them.