[cuco][cudax] Keep cudax cuco up to date with recent cuco changes

During the ongoing migration of cuCollections into cudax we have to keep track of upstream changes that haven't been incorporated into cudax yet.

The upstream cuCollections repo features a number of recent improvements that should be included in the cudax cuco implementation. This issue is meant as a tracker and should be extended when needed.

Compare the upstream cuCollections repository with the cudax cuco implementation to identify recent improvements that are still missing. Extend the tracker with the specific changes found, then verify that the corresponding cudax cuco implementation incorporates them.

Written by the indexing model from the issue text.