-
Notifications
You must be signed in to change notification settings - Fork 1
Open
Description
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
Labels
No labels