Kernels:
Trusted publisher
| __version__ = "0.6.0" | |
| from . import layers | |
| from .ops.gated_delta_rule import chunk_gated_delta_rule, fused_recurrent_gated_delta_rule | |
| from .ops.kda import chunk_kda, fused_recurrent_kda | |
| __all__ = [ | |
| # original functions directly | |
| "chunk_kda", "fused_recurrent_kda", | |
| "chunk_gated_delta_rule", "fused_recurrent_gated_delta_rule", | |
| # layers | |
| "layers", | |
| ] | |