"""ML utilities"""