search

jax-ml / ml_dtypes

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

Refactor ufunc implementations to use variadic template arguments for the inputs. #204

Closed copybara-service[bot] closed 1 month ago

copybara-service[bot] commented 1 month ago

Refactor ufunc implementations to use variadic template arguments for the inputs.

Allows consolidating down to one ufunc implementation for each output arity.

Cleanup only, no functional changes intended.