support for XLA on metal/mps (tensorflow, jax, pytorch)

Question

ngam OP

Created Jun ’22

Replies 2

Boosts 3

Views 1k

Participants 2

I am curious about XLA support. Is this planned for tensorflow-metal/tensorflow-macos?

That would likely work seamlessly for JAX and PyTorch as well.

Answered by Frameworks Engineer in 717983022

Hi @ngam!

Unfortunately we can't comment here any future plans we have. But I will pass forward that there is interest for support for XLA in the developer community.

Boost

Answer 1

Frameworks Engineer OP

Apple

Jun ’22

Accepted Answer

Hi @ngam!

Unfortunately we can't comment here any future plans we have. But I will pass forward that there is interest for support for XLA in the developer community.

0

Answer 2

ngam OP

Jun ’22

Thank you. I understand.

If there is a need for a lengthier explanation, I would be happy to provide one. Both TensorFlow and JAX will likely gain a lot in terms of performance if XLA is supported here. This will likely provide developers (and users) exciting tools to develop their models. It goes without saying that if that could be open-sourced, it will be great. :)

2