Skip to content

pytorch recipe #6

@escorciav

Description

@escorciav

Create your own layer (forward&backward)

(within python)

  • numpy/scipy, check documentation/tutorial.

  • cupy
    For GPU acceleration.
    numpy operations. use dlpack as bridge.
    Porting existing kernel as a string. example

  • numba.
    speed-up cpu forward&backward.
    write cuda kernel within python. example
    suggestions from numba-core-dev.

  • cython
    speed-up CPU forward&backward.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions