Draft: Enable likelihoods to operate over pytrees

Tree-maps and Tree-reduces the likelihood functions to operate on arbitrary tree-like structures.

Merge request reports

Loading