Hey @5had3z, were you able to do inference when you registered grid_sampler? I was able to export a model that uses grid_sample to the ONNX format, but I wasn’t able to do inference with that same model.