Function try_elem_max 

pub fn try_elem_max(
    a: &GpuTensorHandle,
    b: &GpuTensorHandle,
) -> Option<GpuTensorHandle>

Convenience: perform elementwise max via provider if possible; otherwise return None