Real-Only Scipy Special Functions Don't Throw On Complex Input #32104
Description
Description
There are several scipy special lequivalent functions (like polygamma, expi, and expn) that do not take complex input, but don't throw on complex input. We should update these functions to raise an error when the input is complex, or update them to accept complex input.
Related: #6765
Related: #32052
System info (python version, jaxlib version, accelerator, etc.)
jax: 0.4.30
jaxlib: 0.4.30
numpy: 2.0.2
python: 3.9.6 (default, Apr 30 2025, 02:07:17) [Clang 17.0.0 (clang-1700.0.13.5)]
jax.devices (1 total, 1 local): [CpuDevice(id=0)]
process_count: 1
platform: uname_result(system='Darwin', node='Michaels-MBP.lan', release='24.5.0', version='Darwin Kernel Version 24.5.0: Tue Apr 22 19:54:43 PDT 2025; root:xnu-11417.121.6~2/RELEASE_ARM64_T8132', machine='arm64')