OperatorAdapter.jax_expr: Fix casting in recent JAX

Merged Gordian Edenhofer requested to merge fix_op_T_jax_expr into NIFTy_8

Merge request reports