Support for larger memory GPUs

I would welcome this feature as well.