You can have a code using a GPU array library and just differentiate it (which ends up being more flexible / composable).
You can have a code using a GPU array library and just differentiate it (which ends up being more flexible / composable).