Can I use models and architecture that were specifically written for Google TPU?

intel / intel-extension-for-tensorflow

Intel® Extension for TensorFlow*

Other

314 stars 39 forks source link

Can I use models and architecture that were specifically written for Google TPU? #74

Closed sleepingcat4 closed 1 month ago

sleepingcat4 commented 1 month ago

XLA computation is blazingly fast and the models are faster. I looked into Intel repos and there's no support for Jax or XLA. is there any chance we can have an intel extension for Jax?

sleepingcat4 commented 1 month ago

I found openxla!