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

[ml_dtypes] Add int2 and uint2 types. #154

Closed copybara-service[bot] closed 3 months ago

copybara-service[bot] commented 3 months ago

[ml_dtypes] Add int2 and uint2 types.

These are very similar to int4 types in structure.