Module argmax 

Argmax tracking for tropical backward pass (automatic differentiation).

In tropical semirings, the “gradient” of a contraction is determined by which elements “won” the max (or min) comparisons. ArgmaxTracker records the winner indices during the forward pass so that the backward pass can route gradients to the correct elements.

This is the tropical analogue of storing intermediate activations for backpropagation in standard neural network training.

Structs

ArgmaxTracker

Tracks winner indices from tropical forward-pass operations.