You are using an outdated browser. Please upgrade your browser to improve your experience.
The latest news from Google on open source releases, major projects, events, and student outreach programs.
We are excited to announce the first PJRT plugin implementation in Intel Extension for TensorFlow , which seamlessly runs JAX models on Intel ® GPU . The PJRT API simplified the integration, which allowed the Intel GPU plugin to be developed separately and quickly integrated into JAX. This same PJRT implementation also enables initial Intel GPU support for TensorFlow and PyTorch models with XLA acceleration.