Examples
API
References
Apply vjp to recover gradient in reverse mode differentiation.
LinearProblem
prob (ott.problems.linear.linear_problem.LinearProblem) –
f (jax.Array) –
g (jax.Array) –
lse_mode (bool) –
gr (Tuple[jax.Array, jax.Array]) –