Jake Vanderplas

There is some information on jaxprs here: https://jax.readthedocs.io/en/latest/jaxpr.html

What does the `doJax` function contain?

Yes, memory allocation/deallocation should be handle-able through JAX, but I cannot reproduce the issue. For example, this works fine, despite the total size of the created arrays being larger than...

Have you tried benchmarking these? Once compiled to XLA, the `where` based solution shows comparable execution time compared to the straightforward approach: ```python import numpy as np import jax.numpy as...

Hi, thanks for the question. We've just finished copyedits and I'm hoping to push the updated notebooks & re-build the website in the coming weeks. Note that the notebooks on...

The instructions in the updated second edition notebook should work: https://github.com/jakevdp/PythonDataScienceHandbook/blob/d66231454ef753818dc9213c9b5942e067266966/notebooks/03.10-Working-With-Strings.ipynb

Hi @rgommers – I'm having trouble squaring the advice above: > at that point, you can release packages that will work with both 2.0 and 1.X, and hence your own...

Link updated, sorry about that

Ah, thanks. I just found the 1.21 docs via google search; I hadn't seen the 1.26 update.

There are some failures in one of the CI jobs – looks like inf/nan mismatches.