Hi folks,
Thank you so much for the continuous work on XLA. Without it, we would not be able to
bring machine learning to Elixir. In particular, we integrate with Elixir using the PjRT APIs.
Lately, we have been exploring support for PjRT plugins and we have noticed that both IREE and the CUDA plugins for PjRT do not support infeeds/outfeeds. My understanding is that JAX, for example, also moved away from the indeed/outfeed API.
Therefore, before investing further into infeed/outfeed, is it still in the long term plan to support those operations? Should we encourage plugin authors to fill in the missing gaps?