search

JaxGaussianProcesses / GPJax

Gaussian processes in JAX.
https://docs.jaxgaussianprocesses.com/
Apache License 2.0
436 stars 51 forks source link

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

Closed daniel-dodd closed 9 months ago

daniel-dodd commented 9 months ago

Replace all typing using

from jax.random import KeyArray

With 

from jaxtyping import Key