Small code improvements to clean up unneeded optimizations the JIT handles.
Enabled AVX support for the GPU stack.