Skip to content

Releases: sail-sg/autofd

Update code to be compatible with jax 0.4.38

17 Jan 06:31
Compare
Choose a tag to compare
  • fix calls to WrappedFun and lu.transformation
  • add hash to Tracer
  • add to_tangent_aval

v0.0.7

20 Feb 23:11
6284f32
Compare
Choose a tag to compare

Adapt with latest jax changes

v0.0.6

27 Dec 14:44
Compare
Choose a tag to compare

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.

v0.0.5

01 Dec 06:24
e8c9bed
Compare
Choose a tag to compare

Fix the data type for nabla

Changelog

30 Nov 18:07
dd712fd
Compare
Choose a tag to compare
  • Fix support for python < 3.11
  • Minimize dependency

v0.0.3

30 Nov 09:13
Compare
Choose a tag to compare

First release without big glitches.