Jake Vanderplas

Closing due to lack of activity here – feel free to comment here or open another issue if you're still running into problems!

Hi - thanks for the report! JAX itself doesn't pin any `libgrpc` version. It looks like this pinning is likely coming from the `jaxlib` conda-forge package: https://github.com/conda-forge/jaxlib-feedstock. I'm not sure...

It looks like this has been resolved in conda-forge. Thanks!

Thanks for the report – I think this is expected, because XLA flushes subnormal values during operations, because hardware like TPU and some GPU devices do not support them

Hi - thanks for the report! The narrow-width data types like `[u]int4` and `float8_*` are still only experimentally supported, and it's not expected that they will work on all backends....

Looks like a great start! Cross-referencing a couple related issues: #11319 #18398

This looks related to #16116; assigning @sharadmv

I think the logic in the uniform example looks pretty good now. We'd want to apply the same pattern in other cases, and then fix the tests according to my...

Hi - let me know if you're interested in continuing to work on this!

We need to rethink this. I think the desired semantics are: - if the user specifies a dtype, then we should cast inputs to that dtype. - if the user...