kda-neuron-kernels / build /torch-neuron /kda_chunk_step_v2.py

Commit History

v1.5: exact chunked backward (kda_chunk_step_exact_bwd)
6709b3f

Jim Burtoft commited on

v1.2: training / backward support
07e0355

kda-kernel project commited on