Jake Vanderplas

Thanks - will there be a new PyPI release with the fix?

Not sure... it's still not clear to me to what extent we should expect shape polymorphism to be supported in JAX APIs. Do we have those goals documented anywhere? It's...

Yes, I think we should support this. It will involve some reworking of the ``data_transformer`` architecture that's currently being done in #887, and then adding an xarray transformer to the...

Hi - thanks for the report! This is the expected output in `jax.scipy.linalg`. One of the restrictions on JAX APIs that enables compilation and execution at scale is that Python-level...

xref https://github.com/data-apis/array-api/issues/405 JAX explicitly disables this test: https://github.com/jax-ml/jax/blob/40fc6598f96999271a3c19cfaab6f02579c003d6/tests/array_api_skips.txt#L3-L4 We've followed NumPy here and are waiting for a resolution of the above issue.

Thanks for the clarification! It sounds like building against `numpy~=2.0` is the right solution then: this will accept any version `2.X`, but will not accept `1.X` (and will also reject...

Some history here: "How to think in JAX" came first, but when we reorganized the getting started docs last year, we wrote "key concepts" and planned on deleting "how to...

> Related note: > > ... Thanks @carlosgmartin – I think that would be appropriate to discuss in another issue. This issue is about edge-case behavior in currently defined APIs,...

As we discussed, the structure of the left-nav dropdown is still not exactly how we'd like it, but we can land and iterate. Thanks for working on this!

I don't think `materialize` would be useful for JAX. In that case, it would either be a no-op (during eager execution) or an unconditional error (during traced execution). When JAX...