These functions look like the most obvious candidates for this library from scikit-learn: https://github.com/scikit-learn/scikit-learn/blob/439ea045ad44e6a09115dc23e9bf23db00ff41de/sklearn/utils/_array_api.py#L931-L982

cc @OmarManzoor @ogrisel @betatim