Hi, looks like a nice project. Could you wrap it into an importable package to be used? I.e. using similar function calling: fit = umap.UMAP(n_components=2, random_state=42) x = fit.fit_transform(data) Would be great! Thanks