mirror of
https://github.com/ROCm/jax.git
synced 2025-04-24 23:46:05 +00:00

Call XLA's sqrt instead of defining sqrt to be x**0.5. The two have different behaviors for infinite inputs. Incorporate improvements to acos, sinh, cosh, asinh, and acosh that have previously been made to the versions in the XLA C++ client libraries.