some fix for scan when computing gradients wrt to shared variables that where not passed to the inner function (added a new test in test_scan.py