Skip to content
Snippets Groups Projects
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
No related branches found
No related tags found
Checking pipeline status
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