jax.random.KeyArray deprecated

stanford-crfm / haliax

Named Tensors for Legible Deep Learning in JAX

Apache License 2.0

140 stars 9 forks source link

Closed jennifgcrl closed 2 months ago

jennifgcrl commented 4 months ago

As per jax changelog, from jax 0.4.24 on jax.random.KeyArray is no longer, so haliax fails to import with

AttributeError: module 'jax.random' has no attribute 'KeyArray'

dlwh commented 4 months ago

oops, ok, will fix soon!

dlwh commented 2 months ago

i fixed this at some point