The CustomDist API has been significantly changed between v3 and v5. For you this is good news, because it looks like your likelihood function is written entirely of pytensor ops whose logp graph can be automatically determined. See here for what appears to be a similar case
1 Like