Utilities for ML libraries in python