jaxify
================

.. py:module:: jaxns.framework.jaxify

.. rubric:: :code:`jaxns.framework.jaxify`


.. rubric:: Module Contents

.. py:function:: jaxify_likelihood(log_likelihood, vectorised = False)

   Wraps a non-JAX log likelihood function.

   :param log_likelihood: a non-JAX log-likelihood function, which accepts a number of arguments and returns a scalar
                          log-likelihood.
   :param vectorised: if True then the `log_likelihood` must handle batched inputs, i.e. each input will receive a common
                      set of batched dimensions which the function must handle.

   :returns: A JAX-compatible log-likelihood function.