Using the noise parameters sol.W.W from an SDE solution throws an error when taking gradients

Thank you both for comments and suggestions!