TrainingExecutionContext
Extension of ExecutionContext that supports gradient recording and backward pass. This interface allows high-level training utilities to remain backend-agnostic while leveraging autograd capabilities when available.