Support writing custom NumPy ufuncs in Rust #399
Description
Add support for writing custom NumPy ufuncs in Rust.
Discussed in #398
Originally posted by mhostetter October 29, 2023
Is it possible to write a custom NumPy ufunc in Rust and then apply it on arrays (with the requisite broadcasting) from either Rust or Python?
My application is writing finite field arithmetic ufuncs. I previously implemented this in Python and wrapped in Numba. I'm curious if I can replicate in Rust and achieve better performance.
Forgive the question if it's obvious. I didn't find an answer in documentation.
Metadata
Metadata
Assignees
Type
Projects
Milestone
Relationships
Participants
Activity
kngwyu commentedon Oct 30, 2023
I think this is possible through the unsafe raw NumPy API we are exposing now, but we haven't provided a higher-level Rust API for achieving this. This is not our intention, but we just haven't received an urgent request. Definitely worth considering to implement that, but I'm not sure how difficult it is.