Feature Type

  • [ ] Adding new functionality to pandas

  • [ ] Changing existing functionality in pandas

  • [ ] Removing existing functionality in pandas

Problem Description

I wish I could use pandas on top of JAX instead of NumPy.

Feature Description

Enable selectively to use JAX as the dependency for pandas.

pip3 install pandas[jax]

Alternative Solutions

Use JAX by default.

Additional Context

No response

Comment From: phofl

This needs more context and explanations (advantages, disadvantages, ...). This would require major changes in pandas internals.

Comment From: rinarakaki

Or It's much better if pandas only depends on Apache Arrow and can be installed without NumPy.

Comment From: phofl

Firstly, NumPy Is a required dependency of PyArrow

Secondly, this (if we would want to do this) is years away and would mean a major refactor. Please provide more context, otherwise I'll close this.

Comment From: jbrockmendel

We suggest implementing a Jax-backed ExtesnionArray

Comment From: rinarakaki

@pandas-dev Why you deleted my comment?

Comment From: MarcoGorelli

@rnarkk

  1. this isn't an advertisment booth
  2. please don't tag all of pandas-dev