jax

55a9de3c - [si_vjp] fix bugs around symbolic zeros

[si_vjp] fix bugs around symbolic zeros * fix leaking of internal symbolic zeros in returned cotangents * fix a bug around symbolic zero output tangents