Need to write more tests regarding differentiation (in examples maybe)

[theogf]

There can be differentiation issues (with Zygote) regarding the auxiliary variables, especially with expected_logtilt. #20 Solved this problem for now with a relatively generic solutions but more tests are needed.