Mellon is a non-parametric density estimator based on the NearestNeighbors distribution.
To install with pip you can run:
pip install mellon
Please read the documentation or use this basic tutorial notebook.
import mellon
import numpy as np
X = np.random.rand(100, 10) # 10-dimensional state representation for 100 cells
Y = np.random.rand(100, 10) # arbitrary test data
model = mellon.DensityEstimator()
log_density_x = model.fit_predict(X)
log_density_y = model.predict(Y)