jax

4216f8fa - Accelerate deprecation of legacy JAX FFI calling convention.

Accelerate deprecation of legacy JAX FFI calling convention. In https://github.com/jax-ml/jax/pull/24370, `ffi_call` was updated to return a callable, and the original calling convention was deprecated. This change is part of the deprecation cycle for this calling convention. PiperOrigin-RevId: 708424223