I’ve read that the Coral Edge TPU works with Nx through tflite-elixir, but that means it won’t work with Bumblebee. Are there any consumer TPUs that work directly?
Technically EXLA should support TPU, but we don’t actively test it on TPUs, so you’d need to try and see if it runs.
