jax-ml / jax-ml/ml_dtypes

[RFC] Change float8_e5m2 dtype descr from <f1 to <V1

Open
#216 3 comments 0 reactions 1 assignee Claimed by @hawkinsp View on GitHub
Dominant language
C++
Stars
361
Forks
60
Avg merge
18h 41m
Merged PRs (30d)
3

Description

All data types defined in [ml_dtypes/_src/dtypes.cc](https://github.com/jax-ml/ml_dtypes/blob/main/ml_dtypes/_src/dtypes.cc) are assigned `kNpyDescrKind = 'V'`, except for `float8_e5m2`.

### Why is this an issue?

This discrepancy affects the saving and loading of arrays using .npy or .npz file formats. Specifically, numpy.load() can successfully load an .npy file if its header contains `'descr': '

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.