search

inducer / arraycontext

Choose your favorite numpy-workalike!
6 stars 11 forks source link

jax: change DeviceArray to jnp.ndarray #211

Closed matthiasdiener closed 1 year ago

matthiasdiener commented 1 year ago

See e.g. https://github.com/inducer/pytato/actions/runs/3695907762/jobs/6258890619

Background: https://jax.readthedocs.io/en/latest/jax_array_migration.html

Needs:

inducer commented 1 year ago

Thanks!