Add test script for CAE gradient
Fix CAE so it uses a linear decoder and reconstruction error
More updates, reconstruction gradient now correct