Operator Registration Routine

XLACustomOp	Creating a XLA custom call operator.
defjvp	Define JVP rules for any JAX primitive.
register_general_batching