Kernels:

jburtoft
/

kda-neuron-kernels

Kernel card Files Files and versions

kda-neuron-kernels / build /torch-neuron

238 kB

Ctrl+K

Ctrl+K

4 contributors

History: 8 commits

Jim Burtoft

v1.6: kernel-repo distribution fix

eacd70d about 9 hours ago

__init__.py
8.12 kB
v1.5: exact chunked backward (kda_chunk_step_exact_bwd) about 16 hours ago
kda_autograd.py
12.4 kB
v1.3: fused multi-chunk backward 2 days ago
kda_chunk_bwd.py
29.8 kB
v1.5: exact chunked backward (kda_chunk_step_exact_bwd) about 16 hours ago
kda_chunk_step_v2.py
19.5 kB
v1.5: exact chunked backward (kda_chunk_step_exact_bwd) about 16 hours ago
kda_fused_chunked_bwd.py
31.5 kB
v1.5: exact chunked backward (kda_chunk_step_exact_bwd) about 16 hours ago
kda_fused_chunked_fwd.py
20.8 kB
v1.5: exact chunked backward (kda_chunk_step_exact_bwd) about 16 hours ago
kda_recurrent_bwd.py
16.1 kB
v1.5: exact chunked backward (kda_chunk_step_exact_bwd) about 16 hours ago
kda_recurrent_fwd_v2.py
6.48 kB
v1.5: exact chunked backward (kda_chunk_step_exact_bwd) about 16 hours ago
metadata.json
190 Bytes
v1.6: kernel-repo distribution fix about 9 hours ago
nki_kda.py
12.6 kB
v1.5: exact chunked backward (kda_chunk_step_exact_bwd) about 16 hours ago
nki_kda_chunked.py
20.6 kB
v1.5: exact chunked backward (kda_chunk_step_exact_bwd) about 16 hours ago
nki_kda_chunked_exact.py
15.3 kB
v1.5: exact chunked backward (kda_chunk_step_exact_bwd) about 16 hours ago
nki_kda_chunked_exact_bwd.py
24.8 kB
v1.5: exact chunked backward (kda_chunk_step_exact_bwd) about 16 hours ago
nki_kda_chunked_exact_multihead.py
12.7 kB
v1.5: exact chunked backward (kda_chunk_step_exact_bwd) about 16 hours ago
nki_kda_decode_batch.py
6.82 kB
v1.5: exact chunked backward (kda_chunk_step_exact_bwd) about 16 hours ago