Jax: Using JAX with Pandas

Created on 8 May 2020  路  3Comments  路  Source: google/jax

Hey,
Is there any way to use JAX to speed up and reduce memory-usage of data processing pipelines in pandas ?

enhancement

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!

All 3 comments

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,

Was this page helpful?
0 / 5 - 0 ratings