Port `max` kernel to structured kernels. (#61449)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/61449
Tracking issue: #55070
Test Plan: Imported from OSS
Reviewed By: saketh-are
Differential Revision: D29741714
Pulled By: bdhirsh
fbshipit-source-id: 6c8c17d20f578ab0af8a969d103a19ccd8d51842