Skip to content
Unverified Commit 4c8b4314 authored by Matthias Diener's avatar Matthias Diener Committed by GitHub
Browse files

jax: change DeviceArray to jnp.ndarray (#211) 

* limit JAX version due to CI errors

* rename DeviceArray -> Array

* import as JAXArray

* Revert "limit JAX version due to CI errors"

This reverts commit 9b89d14c3ee8eee1f5ded8b9fc8eee31f4c3fb85.

* change to jax.numpy.ndarray
parent 1199dc2e
Pipeline #372348 passed with stage
in 1 hour and 14 seconds
Hide whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment