Is it possible to optimize on a subset of the data?

[sebamenabar]

Hi, I'm doing an interactive scatter plot where you can zoom in or show subsets of the points. I perform an initial optimization with all the data, and then I'd like to optimize the embeddings for a subset of the data, but also beggining from the embeddings found in the initial optimization.

Any recommendation you have to achieve this?

P.S. Thanks for the library, it's super cool