Added PyTorch link and unit tests for argmax#900

Closed

HangenYuu wants to merge 1 commit intopymc-devs:mainfrom HangenYuu:torch_argmax

+40

Commits on Jul 8, 2024

Started working on argmax for torch
HangenYuu
committed