Skip to content

Use numba to accelerate numerical CPU code #1972

@aseembits93

Description

@aseembits93

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!

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or request

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions