transformers

f19d018b - Revert "Update deprecated Jax calls (#35919)" (#36880)

Revert "Update deprecated Jax calls (#35919)" (#36880) * Revert "Update deprecated Jax calls (#35919)" This reverts commit f0d5b2ff04e1354d32beac70984adcc8100352a0. * Revert "Update deprecated Jax calls (#35919)" This reverts commit f0d5b2ff04e1354d32beac70984adcc8100352a0. * udpate