[Task]: Make `Leaf dynamics` differentiable and efficient

[SarahAlidoost]

The module Leaf dynamics is already differentiable and efficient wrt SPAN and TDWI. Implement the same changes for remaining parameters. See explanations and requirements in #39. and see this discussion.