Method logsumexp

ot/backend.py:1621–1622

(self, a, axis=None, keepdims=False)

        return jnp.log(a)

    def logsumexp(self, a, axis=None, keepdims=False):
        return jspecial.logsumexp(a, axis=axis, keepdims=keepdims)

    def sqrt(self, a):
        return jnp.sqrt(a)