Function tropical_matmul_with_argmax_with_workspace 

pub fn tropical_matmul_with_argmax_with_workspace<T: TropicalWithArgmax<Index = u32> + KernelDispatch>(
    a: &[T::Scalar],
    m: usize,
    k: usize,
    b: &[T::Scalar],
    n: usize,
    workspace: &mut GemmWorkspace<T::Scalar>,
) -> GemmWithArgmax<T>

Tropical matrix multiplication with argmax and reusable packing storage. Output values and indices are newly allocated; workspace is reused across calls.