nemos.observation_models.PoissonObservations.check_inverse_link_function

static PoissonObservations.check_inverse_link_function(inverse_link_function)

Check if the provided inverse_link_function is usable.

This function verifies if the inverse link function:

1. Is callable

2. Returns a jax.numpy.ndarray

3. Is differentiable (via jax)

Parameters:

inverse_link_function (Callable) – The function to be checked.

Raises:

TypeError – If the function is not callable, does not return a jax.numpy.ndarray, or is not differentiable.