def f(x):
y = jnp.sin(x)
debugger.breakpoint(stdin=stdin, stdout=stdout)
return y
def g(x):
y = f(x) * 2.
debugger.breakpoint(stdin=stdin, stdout=stdout, ordered=True)
return jnp.exp(y)
def f(x):
y = x + 1.
debugger.breakpoint(stdin=stdin, stdout=stdout)
return 2. * y
def f(x):
y = x + 1.
debugger.breakpoint(stdin=stdin, stdout=stdout, ordered=True)
return y