TensorFlow Pluggable Device

tenstorrent / tt-metal

:metal: TT-NN operator library, and TT-Metalium low level kernel programming model.

Apache License 2.0

478 stars 78 forks source link

TensorFlow Pluggable Device #15138

Open mbahnasTT opened 5 days ago

mbahnasTT commented 5 days ago

Is your feature request related to a problem? Please describe. TensorFlow Pluggable Device

Describe the solution you'd like

https://blog.tensorflow.org/2021/06/pluggabledevice-device-plugins-for-TensorFlow.html
https://www.tensorflow.org/install/gpu_plugins
https://github.com/tensorflow/community/blob/master/rfcs/20200624-pluggable-device-for-tensorflow.md

zzigler-tt commented 17 hours ago

The FR would fall under @ayerofieiev-tt for development, although more information wrt customer expectations, commitments, and deliverable requirements for TT are needed in order to full scope this item.