DifferentialEquation.prepare_node(node, storage_map, compute_map, impl)#

Make any special modifications that the Op needs before doing Op.make_thunk().

This can modify the node inplace and should return nothing.

It can be called multiple time with different impl values.


It is the Op’s responsibility to not re-prepare the node when it isn’t good to do so.