Jake Vanderplas

Hi - Jax does not have any built-in support yet for a pandas-like API, but it is an interesting idea. I'll leave this as an open issue!

I've been playing with this idea a bit; I pushed my current work-in-progress to https://github.com/google/jax/pull/12329

Thanks for the question! I'm assigning @mattjj because he knows some of the context of discussions around these kinds of ideas.

Mine is an intel macbook

`DeviceArray` does not behave this way, but we want `Array` to behave this way. If we lower `Array` to C++, we should add an equivalent metaclass in the C++ definition,...

I've found a couple examples of overriding metaclasses / `__instancecheck__` within pybind11; one example is in torch: https://github.com/pytorch/pytorch/blob/31ef8ddb8c4467f5b8698ef1eb9bb8bab7056855/torch/csrc/tensor/python_tensor.cpp#L149-L177

Replaced by https://github.com/google/jax/pull/12300

I had been making some effort to land independent parts of this separately in order to make the change less extensive (e.g. https://github.com/google/jax/pull/12370). Do you think that's worth doing here?...

I'm more worried about the things I haven't thought of... and changes like #12370 are truly independent – they're correct whether or not we land the rest of this, so...

Thanks for raising this: I think at this point the best practices generally involve using `Any` for anything non-trivial (particularly pytrees and arrays), owing to some general missing features in...