Hey,
Is there any way to use JAX to speed up and reduce memory-usage of data processing pipelines in pandas ?
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!
In the meantime, you can also read pandas dataframes directly as a tf.data.Dataset (which would be compatible with JAX).
Thank you for the feature request!
@jakevdp @mattjj @dynamicwebpaige - any updates on this enhancement? thanks,
Most helpful comment
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!