search

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

Adjust ml_dtypes to support NumPy 2.0 dtype registration APIs #136

Closed copybara-service[bot] closed 7 months ago

copybara-service[bot] commented 7 months ago

Adjust ml_dtypes to support NumPy 2.0 dtype registration APIs

The signature of PyArray_RegisterDataType changes in NumPy 2.0, along with the lifetime semantics of its arguments.