Method solve

ot/backend.py:1828–1829

(self, a, b)

        return results

    def solve(self, a, b):
        return jnp.linalg.solve(a, b)

    def trace(self, a):
        return jnp.diagonal(a, axis1=-2, axis2=-1).sum(-1)