In the end I had to combine the last two versions. For jacobians:
- We want to split according to the FunctionView (e.g. Stokes).
- We don't want to split according to gradients (e.g. sumfactorization).