Hello! I think this can be done but I would not advise you to do it with this particular algorithm.
I think you would be interested in the work of Jatin Chowdhury[1] as well as the work of Christian J. Steinmetz[2].
[1].https://arxiv.org/pdf/2009.02833.pdf [2] https://arxiv.org/abs/2102.06200