nanodl
nanodl copied to clipboard
Jax version
With the latest jax=='0.4.25'
and jaxlib=='0.4.25'
I get:
import nanodl
# AttributeError: module 'jax.random' has no attribute 'KeyArray'