"""Machine-learning inference wrappers used by the ml-worker."""