Update Compat Bounds for NNlib

[jacobadenbaum]

I've updated the compat bounds for NNlib to allow v0.9

I've also fixed an issue with the GPU tests where the test environment was erroring because it required an old version of CUDA and NNlibCUDA (which has been deprecated in favor of a CUDA extension in NNlib).

Please note: I have not been able to verify that the GPU tests run, since I do not have access to CUDA on my development machine.

[jacobadenbaum]

I'll note: although it looks like the CI tests for this are failing, I don't think that has anything to do with this PR. It would be great if a maintainer could look at the CI setup for this to figure out what the problem is, since its not something that I think I can resolve here