Interop with Numba / JAX?

[EricCousineau-TRI]

This is very far left field, but it'd be interesting if there were a way for pybind11 and numba to interoperate.

One scenario might be if a pybind signature overload could be detected during @numba.jit, and if numba could detect pybind types, get the underlying C[++] data, and strip out marshaling normally done by pybind, and reduce the overhead when calling C++ code from Python in loops (related: #1227).

[EricCousineau-TRI]

Adding JAX, as that seems like a potential successor / partner to Numba: https://github.com/google/jax

[CFAndy]

Hi Eric, I'm interested in the inter-operation between pybind11 module and python based callback which is jitted by Numba/JAX.

[EricCousineau-TRI]

Howdy Andy, that's awesome! I won't be able to focus too deeply on this front atm, but can I ask if it's something you'd be willing to prototype and if there's anything I could help point you towards (if you need any resources)?