JaxGaussianProcesses/GPJax

dev: `jax.random`'s `KeyArray` is deprecated - move to `jaxtyping`'s `Key`

daniel-dodd opened this issue · 0 comments

daniel-dodd commented

Replace all typing using

from jax.random import KeyArray

With

from jaxtyping import Key