slayerPytorch icon indicating copy to clipboard operation
slayerPytorch copied to clipboard
verified publisher icon bamsumit

slayerJax

Open never-to-never opened this issue 3 years ago • 1 comments

Hello, I want to use jax to implement slayer, while retaining the cuda accelerated part in slayerCuda.conv and slayerCuda.getSpikes, can you give me a detailed guide?

never-to-never avatar Aug 23 '22 03:08 never-to-never

Hi, I am happy to provide pointers. Before you start implementing, you may want to port Slayer 2.0 instead (https://github.com/lava-nc/lava-dl). It's newer, supports more features, and is actively maintained. If you agree, let's move this discussion to lava-dl.

bamsumit avatar Aug 29 '22 15:08 bamsumit