This make this test executed on the gpu too.

Fixed grad of grad of scan

Fixed tests with script run_tests_in_batch.py

Note sure it makes the file anymore readable, but at least I've tried.

Otherwise the numberic differentiation gradient quickly becomes numerically
unstable.