Backend for loglikelihood calculation

[pawel-czyz]

Overview

Backend for calculating loglikelihood and its derivatives.
It can be based e.g., on LearnMHN package and joblib parallelisation.

Tasks:

• Loglikelihood and gradient for a single genotype and MHN.
• Loglikelihood and gradient for several genotypes and shared MHN.
• Loglikelihood and gradient for several genotypes and several MHNs.

Description

We want to have functions of signatures:

def loglikelihood(genotype: Bool[Array, " M"], theta: Float[Array, "M M"]) -> float:
   ...
   
def gradient(genotype: Bool[Array, " M", theta: Float[Array, "M M") -> Float[Array, "M M"]:
 ...

implementing the loglikelihood and the gradient for a particular genotype.

Apart from that we want to have vectorized versions as described above.

[pawel-czyz]

This is done via #9