jax-ml / ml_dtypes

A stand-alone implementation of several NumPy dtype extensions used in machine learning.
Apache License 2.0
191 stars 25 forks source link

Adept to NumPy change #135

Closed seberg closed 7 months ago

seberg commented 7 months ago

To allow changing the details of NumPy DTypes/descriptor struct, I need to slightly backwards compatible changes to the registration.

Since ml_types is probably the larger open source place, just a heads up: You will need to do some slight modifications: https://numpy.org/devdocs/reference/c-api/array.html#c.PyArray_RegisterDataType (missing colon in the doc rendering, but I think you can get the idea).

jakevdp commented 7 months ago

Thanks! Fixed by #136