Danis
Danis
@dgasmith Unfortunately, not sure what tests to add.
It might be useful: https://github.com/jax-ml/jax/pull/3856#issuecomment-663746553
When `x > 26`, we can compute the asymptotic expansion as `x -> +infinity`: ``` erfcx(x) = 1/sqrt(π) * (1/x - 1/(2x³) + 3/(4x⁵) - 15/(8x⁷)) ``` For `x >...
Gubaydullin Danis [email protected]
@MythicArrow It seems that `prelu` is equivalent to `leaky_relu`?
You can pass a JAX array as the second argument to `leaky_relu`, and JAX will have no issues computing gradients through it.