nanodl icon indicating copy to clipboard operation
nanodl copied to clipboard

Jax version

Open afrendeiro opened this issue 11 months ago • 1 comments

With the latest jax=='0.4.25' and jaxlib=='0.4.25' I get:

import nanodl
# AttributeError: module 'jax.random' has no attribute 'KeyArray'

afrendeiro avatar Mar 12 '24 08:03 afrendeiro