Implement Jax CPU/GPU callbacks with XLA's FFI.

- Change 4 of 4 addressing #3 in jax-ml/jax#25842.

PiperOrigin-RevId: 729641154

Author: danielsuo

third_party/xla/xla/python/xla_client.py

@@ -50,7 +50,7 @@
 
 # Just an internal arbitrary increasing number to help with backward-compatible
 # changes. In JAX, reference this via jax._src.lib.xla_extension_version.
-_version = 316
+_version = 317
 
 # Version number for MLIR:Python components.
 mlir_api_version = 58