Releases: sail-sg/autofd
Releases · sail-sg/autofd
Update code to be compatible with jax 0.4.38
v0.0.7
v0.0.6
Add braket operator.
braket(f, g)
is equivalent to o.integrate(o.numpy.conj(f) * g)
.
However, the former is preferred because currently if we use functional gradient
jax.grad(lambda f: o.integrate(o.numpy.conj(f), f))
It somehow results in redundant computation paths that is not optimised away by XLA.