"""Diagonal Kalman Delta Network (KDN) token mixer and recurrences."""

from tiny_gdn.kdn.layer import SUPPORTED_HEAD_DIMS, DiagonalKalmanDeltaNet

__all__ = ["SUPPORTED_HEAD_DIMS", "DiagonalKalmanDeltaNet"]