pytorch / torcharrow

High performance model preprocessing library on PyTorch
https://pytorch.org/torcharrow/beta/index.html
BSD 3-Clause "New" or "Revised" License
647 stars 79 forks source link

[Arrow] bf16 support in arrow #521

Open changhiskhan opened 1 year ago

changhiskhan commented 1 year ago

🚀 The feature, motivation and pitch

We're working on supporting bf16 in lance format, which will be presented as a bf16 extension type in Arrow (see PR for details: https://github.com/lancedb/lance/pull/1002).

We're considering contributing the Arrow extension type as a canonical extension to the Arrow project. Would this be of interest / value to the pytorch community? How would this be integrated into the pytorch library?

If y'all are interested, we could start a discussion thread on Arrow mailing list and chat together?

Alternatives

No response

Additional context

No response

cc @SsnL @VitalyFedyunin @ejguan @dzhulgakov