Reverted chunk_based_bmu_find from method to stateless function to allow it to be pickled and run in parallel by joblib