Method allclose

ot/backend.py:1781–1782

(self, a, b, rtol=1e-05, atol=1e-08, equal_nan=False)

        return a

    def allclose(self, a, b, rtol=1e-05, atol=1e-08, equal_nan=False):
        return jnp.allclose(a, b, rtol=rtol, atol=atol, equal_nan=equal_nan)

    def dtype_device(self, a):
        if self.jax_new_version: