Implement simulation-time encoding tying of grid-equivariant ops in aimet-onnx