Closed NeilGirdhar closed 4 months ago
This is intended to be a temporary solution until Jax (ideally) makes these arrays into Jax arrays. I don't see any reasonable way for clients to work around this unless this is fixed on the library side.
(Error appears to be unrelated to this pull request.)
This looks good. Can you add a test.
Great, I added a test.
The tests failure should be fixed by my suggestion, and the test suite failure should be fixed by https://github.com/data-apis/array-api-compat/pull/131.
(Done.)
See https://github.com/google/jax/issues/20620.