File size: 208 Bytes
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"]