Status of OpenXLA PJRT plugin for Apple Metal

Machine Learning & AI General
joelberkeley OP
Created 3w
Replies 2
Boosts 0
Views 204
Participants 2

The OpenXLA PJRT plugin used by JAX is missing features, such as 64-bit floating point. Is there some way to track the progress of these features?

Replies  2
Boosts  0
Views  204
Participants  2
joelberkeley OP
3w

Here https://github.com/jax-ml/jax/issues?q=is%3Aissue+is%3Aopen+metal

0
DTS Engineer OP
Apple
2w

Hi,

Are you referring to the tensorflow-metal plug-in?

0
Status of OpenXLA PJRT plugin for Apple Metal
First post date Last post date
 
 
Q