proc logsumexp[T: SomeFloat](t: Tensor[T]): T

proc stable_softmax[T](x, max, sumexp: T): T {.noSideEffect, inline.}

proc streaming_max_sumexp[T](t: Tensor[T]): tuple[max: T, sumexp: T] {.
    noSideEffect, inline.}

proc streaming_max_sumexp[T](t: Tensor[T]; axis: int): Tensor[
    tuple[max: T, sumexp: T]] {.noinit.}