Bringing PyTorch Monarch to AM... Note