jax
95e3fcdf - Merge pull request #4438 from hawkinsp:gelu

Commit

5 years ago

Merge pull request #4438 from hawkinsp:gelu PiperOrigin-RevId: 335063459

References

#4438 - Adds an approximate=... keyword argument to jax.nn.gelu to select bet…

Author

a-googler

a-googler

Parents

Loading