Large arrays of PRNG keys cannot be deferred

Hi!

First, thanks for this amazing tool!

I noticed that your tool fails to render large PRNG key arrays, see:

Clicking on the arrow reveals the following traceback:

```

```

I think the error is similar to https://github.com/google-deepmind/treescope/issues/52, but here applies to generating the summary of an array, which does not occur for smaller arrays:

*Notes:* I am using `treescope==0.1.10` and `jax==0.7.1`.