[BUG] CUDA random crashes on large sizes

```python
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
```