More callbacks

[sebffischer]

Go through the callback list in luz and see what we also want: https://mlverse.github.io/luz/reference/index.html#callbacks