IMPORTANT: To view this page as Markdown, append `.md` to the URL (e.g. /max/get-started.md). For the complete documentation index, see llms.txt.

Skip to main content

For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).

Mojo module

argmaxmin_gpu

Provides a GPU argmax/argmin kernel that wraps top-k with K=1 for inner-dimension reductions.

Functions

argmax_gpu: Computes the indices of the maximum values along the inner-most dimension of the input tensor on the GPU.
argmaxmin_gpu: Wraps the Top-K GPU kernel with K=1 to perform argmax on the inner-most dimension.
argmin_gpu: Computes the indices of the minimum values along the inner-most dimension of the input tensor on the GPU.

Functions

Edit this page

Edit this page

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!