How feasible would it be to add Coral TPU and USB Accelerator support to Nx?

Elixir Framework Forums Nx Forum
, ,
1

I have a coral.ai Coral TPU USB Accelerator. It is currently unsupported by Nx; I’m thinking about how difficult it would be to add support myself. I know there’s tflite-elixir that could be used for inspiration, also the GPU support. Would someone adding a new TPU need to use C++ or Rust?
I’m still working through the machine learning in elixir book, but thinking that this TPU would be handy attached to a Raspberry Pi.

3 Likes