Total param: 48K
Input X:
array([[ 1964, 12, 32772, ..., 0, 6176, 0], [ 1964, 12, 32772, ..., 0, 6841, 0], [ 1964
