Skip to content

Perform single call to aeppl.factorized_joint_logprob when calling model.logpt #5230

Closed
@ricardoV94

Description

@ricardoV94

For model.logpt we are calling model.datalogpt and model.varlogpt, each of which calls aeppl.factorized_joint_logprob. It should be possible to do a single call to aeppl with all the free and observed model variables, which would save quite a lot of repeated operations on the aeppl side. For some reason this was not working when we tried it in #5166 and #4887.

logp_var = self.varlogpt + self.datalogpt

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions