setting up simple example where arrayfire gradient gives some wonky results