You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
@hawkinsp@jakevdp@seberg@kmaehashi We already had a draft PR to add cupy.complex32 (cupy/cupy#4454) which is made as a legit numpy.dtype via NumPy C APIs (CuPy only uses NumPy dtypes; it does not have its own dtype system). My preference (speaking from NVIDIA perspective) is to add exotic NumPy dtypes to ml_dtypes and encourage all projects to use them. What would be the needed ingredients for me to migrate that PR over to here? Is there a guidance that I can follow along?
We'd need this for 16-bit FFTs in JAX: jax-ml/jax#17954
The text was updated successfully, but these errors were encountered: