5 ms·
XLA (Accelerated Linear Algebra) [1] is likely the library that you'll want to use to code for these machines. TensorFlow, PyTorch, and Jax all support XLA on
by cbarrick 1y ago
XLA (Accelerated Linear Algebra) [1] is likely the library that you'll want to use to code for these machines.
TensorFlow, PyTorch, and Jax all support XLA on the backend.
[1]: https://openxla.org/ https://openxla.org/