Skip to content
Commit 0c1bf0e5 authored by Matthias Diener's avatar Matthias Diener Committed by Andreas Klöckner
Browse files

JAX: Fix DeviceArray for jax 0.4

parent 49586f65