+
    TV-jŸ  ã                   óz   € ^ RI t ^ RIHt ^ RIHt  ! R R]P                  4      t ! R R]P                  4      tR# )é    Nc                   ó^   a a€ ] tR t^	t oV3R lV 3R lltRR ltRV3R lR lltRtVtV ;t	# )	ÚMultiLinearc                ó0   <€ V ^8„  d   QhRS[ RS[ RS[ RR/# )é   Ú
input_dimsÚoutput_dimsÚ	num_headsÚreturnN)Úint)ÚformatÚ__classdict__s   "€Úb/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/mlx_lm/models/mla.pyÚ__annotate__ÚMultiLinear.__annotate__
   s)   ø€ ÷ 
ñ 
¡3ð 
±Sð 
ÁSð 
ÈTñ 
ó    c                ó²   <€ \         SV `  4        \        P                  ! R V,          4      p\        P
                  P                  V) VW2V3R7      V n        R# )g      ð?©ÚlowÚhighÚshapeN)ÚsuperÚ__init__ÚmathÚsqrtÚmxÚrandomÚuniformÚweight)Úselfr   r   r	   ÚscaleÚ	__class__s   &&&& €r   r   ÚMultiLinear.__init__
   sL   ø€ Ü‰ÑÔÜ—	’	˜# 
Õ*Ó+ˆÜ—i‘i×'Ñ'Ø�ØØ¨:Ð6ð (ó 
ˆŽr   c                óz   € V'       d#   WP                   P                  RR4      ,          # WP                   ,          # )é   éÿÿÿÿéþÿÿÿ)r   Úswapaxes©r   ÚxÚ	transposes   &&&r   Ú__call__ÚMultiLinear.__call__   s+   € ßØ—{‘{×+Ñ+¨B°Ó3Õ3Ð3à—{‘{•?Ð"r   c                ó,   <€ V ^8„  d   QhRS[ RS[ RS[/# )r   Ú
group_sizeÚbitsÚmode©r   Ústr)r   r   s   "€r   r   r      s)   ø€ ÷ ñ áðñ ðñ ñ	r   c                óò   € V P                   P                  w  rEp\        WeWAW#4      p\        P                  ! V P                   VVVR 7      vVn         Vn        pV'       d   V^ ,          Vn        V# RVn        V# )©r0   N)r   r   ÚQuantizedMultiLinearr   ÚquantizeÚscalesÚbiases)	r   r.   r/   r0   r	   r   r   Úqlr8   s	   &&&&     r   Úto_quantizedÚMultiLinear.to_quantized   sv   € ð .2¯[©[×->Ñ->Ñ*ˆ	 
Ü!Ø Y¸Dó
ˆô )+¯ªØ�K‰KØØØô	)
Ð%ˆŒ	�2”9˜v÷ "(�F˜1•IˆŒ	Øˆ	ð .2ˆŒ	Øˆ	r   )r   ©T)Úaffine)
Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r   r+   r:   Ú__static_attributes__Ú__classdictcell__Ú__classcell__©r!   r   s   @@r   r   r   	   s#   ù‡ € ÷
ó 
ô#÷÷ ò r   r   c                   óH   a a€ ] tR t^-t oV3R lV 3R lltRR ltRtVtV ;t# )r5   c                ó>   <€ V ^8„  d   QhRS[ RS[ RS[ RS[ RS[ RS[/# )r   r   r   r	   r.   r/   r0   r1   )r   r   s   "€r   r   Ú!QuantizedMultiLinear.__annotate__.   sG   ø€ ÷ ñ áðñ ðñ ð	ñ
 ðñ ðñ ñr   c                óf  <€ \         S
V `  4        W@n        WPn        W`n        \
        P                  ! ^V,          4      p\        P                  P                  V) VW2V3R7      p\        P                  ! W„WVR7      vV n        V n        p	V	'       d
   V	^ ,          MRV n        V P                  4        R# )r$   r   r4   N)r   r   r.   r/   r0   r   r   r   r   r   r6   r   r7   r8   Úfreeze)r   r   r   r	   r.   r/   r0   r    r   r8   r!   s   &&&&&&&   €r   r   ÚQuantizedMultiLinear.__init__.   s–   ø€ ô 	‰ÑÔà$ŒØŒ	ØŒ	ô —	’	˜!˜j�.Ó)ˆÜ—‘×"Ñ"Ø�ØØ¨:Ð6ð #ó 
ˆô
 -/¯KªKØ ô-
Ð)ˆŒ�T”[ 6÷ $*�f˜Q–i¨tˆŒà�‰Žr   c                ó´   € \         P                  ! VV R ,          V R,          V P                  R4      VV P                  V P                  V P
                  R7      # )r   r7   r8   )r7   r8   r*   r.   r/   r0   )r   Úquantized_matmulÚgetr.   r/   r0   r(   s   &&&r   r+   ÚQuantizedMultiLinear.__call__K   sJ   € Ü×"Ò"ØØ��NØ˜•>Ø—8‘8˜HÓ%ØØ—‘Ø—‘Ø—‘ô	
ð 		
r   )r8   r/   r.   r0   r7   r   r<   )	r>   r?   r@   rA   r   r+   rB   rC   rD   rE   s   @@r   r5   r5   -   s   ù‡ € ÷ó ÷:

ô 

r   r5   )	r   Úmlx.coreÚcorer   Úmlx.nnÚnnÚModuler   r5   © r   r   Ú<module>rV      s1   ðó å Ý ô!�"—)‘)ô !ôH(
˜2Ÿ9™9ö (
r   