02720c5
1
2
3
4
5
6
"""Diagonal Kalman Delta Network (KDN) token mixer and recurrences.""" from tiny_gdn.kdn.layer import SUPPORTED_HEAD_DIMS, DiagonalKalmanDeltaNet __all__ = ["SUPPORTED_HEAD_DIMS", "DiagonalKalmanDeltaNet"]