| """Diagonal Kalman Delta Network (KDN) token mixer and recurrences.""" | |
| from tiny_gdn.kdn.layer import SUPPORTED_HEAD_DIMS, DiagonalKalmanDeltaNet | |
| __all__ = ["SUPPORTED_HEAD_DIMS", "DiagonalKalmanDeltaNet"] | |
| """Diagonal Kalman Delta Network (KDN) token mixer and recurrences.""" | |
| from tiny_gdn.kdn.layer import SUPPORTED_HEAD_DIMS, DiagonalKalmanDeltaNet | |
| __all__ = ["SUPPORTED_HEAD_DIMS", "DiagonalKalmanDeltaNet"] | |