jax

2bf9d5aa - loosen tolerance on tanh grad test