Use numba to accelerate numerical CPU code #1972
Description
Search before asking
- I have searched the Inference issues and found no similar feature requests.
Description
Hi all, I was wondering if there's a possibility to use numba to accelerate CPU code in this repository? With minor refactors, we can see massive performance boosts by adding numba's @jit decorator. Hope to hear your thoughts. Best,
Use case
No response
Additional
No response
Are you willing to submit a PR?
- Yes I'd like to help by submitting a PR!