related: https://github.com/verdverm/pge-jax

A deterministic algorithm for Symbolic Regression, which I recently ported to JAX