jax

42258571 - add special value grad tests, sinh failing