A neat feature of Keras is that its `keras.ops` API follows the NumPy API syntax exactly. And since it runs on JAX, PyTorch, TensorFlow, and NumPy itself, you can use Keras to write GPU-accelerated NumPy code that runs with any framework.

→ View original post on X — @fchollet