[BUG] CUDA random crashes on large sizes

[awni]

import mlx.core as mx

a = mx.random.uniform(shape=(1024, 1024, 1024, 3))
mx.eval(a)

Fails with:

RuntimeError: cudaGraphAddKernelNode(&node, graph_, NULL, 0, &params) failed: invalid argument