Closed LisaCoiffard closed 6 months ago
Related issues: N/A
Fix a bug in types.py script, jax.random.KeyArray is deprecated in JAX and should be replaced by jax.Array.
types.py
jax.random.KeyArray
jax.Array
n/a
Related issues: N/A
Fix a bug in
types.py
script,jax.random.KeyArray
is deprecated in JAX and should be replaced byjax.Array
.Checks
Future improvements
n/a