[Feature]: Support for Full Time-Derivative Prediction (the Right Hand Side of System of ODEs) with Symbolic Regression

[rucky96]

Feature Request

Hi,

Thank you for your continued work on PySR! I’ve been interested lately in using PySR to model complex physical systems, specifically working on symbolic regression for time series.

I'm trying to predict the full set of time derivatives ( \frac{dV_i}{dt} ) for all variables ( V_i ). This is just a system of ODEs ( \frac{dV}{dt}= f(V) ). For example, a lorenz system, Rossler, harmonic oscillator... The current symbolic regression approach seems to handle one variable at a time, but for this case, the variables are interrelated. Thus, I need to handle the entire set of time derivatives across all modes simultaneously to apply (for example) conservation laws that I can include in the loss function.

Would it be possible to access to the whole prediction of the vector ( V)? Do you think this could be interesting to explore further? I'd be happy to discuss potential ideas and how this might fit into PySR’s future development.

Looking forward to your thoughts! Thanks again for all your efforts!

Best,
David