7GB 샤드가 640개면 4.4TB 가중치인데, 왜 포스트에는 8TB라고 적혀있지?
텐서 형태를 보면 대충 이럼
ffn.gate.bias_v1 [384]
seq.value_scale [16, 160]
seq.out_a.scale [256, 128]
seq.out_b.scale [160, 256]
seq.query_a.scale [40, 160]
a와 b의 페어링을 통한 쿼리는 병렬 어텐션 경로일 수 있음. 스케일조차 차원이 다르다는 점도 흥미롭네.
스케일은 양자화 스칼라일 수 있고 0C_28FF는 그들의 커스텀 양자화 포맷일지도 모르며, 그들의 기술은 위와 같은 스칼라 행렬을 사용할 것임.
gate bias는 F32로 남겨둔 걸 보니, 스칼라와 함께 기존의 정밀도(F16/BF16/F8/E4M3 등)를 사용하지 않는 것 같음. 그래서 독자적인 기술로 양자화했다는 신호라고 추측함.
만약 8비트라면 4.4T 모델 사이즈 이상이겠지. 8비트는 이미 추론에 매우 훌륭하니까, 커스텀 dtype 투자는 4비트로 밀어붙여서 거의 9T 모델까지 키우려는 것일 수도 있음. 사실이라면 엄청난 일이지. 연구소들이 9T 사이즈까지 가는 건 들어본 적 없지만, 수천억 달러를 날리고 있으니 비용 측면에서는 그리 놀랍지 않을 수도 있겠네.