For large models, sampling, calibration and optimization can become challenging with TorchDiffEq, How much effort would be required to enable the option to run the chirho-diffeqpy effect handler when running sample/calibrate/optimize?
Can you break it down into subtasks?