+
    –°³jã  ã                   ód   a € R t0 t ^ RIt^ RIHu Ht ^ RIHt ^ RIHt . t	] ^ k  ! R R4      t
R# )é    N)ÚTensor)Ú2_scripted_functional_optimizer_deprecation_warningc                   óH   a € ] tR t^t o RV 3R lR lltV 3R lR ltRtV tR# )Ú_FunctionalAdamaxc                óp   <€ V ^8„  d   QhRS[ S[,          RS[RS[S[S[3,          RS[RS[RS[RS[RS[/# )	é   ÚparamsÚlrÚbetasÚepsÚweight_decayÚforeachÚmaximizeÚ_allow_empty_param_list)Úlistr   ÚfloatÚtupleÚbool)ÚformatÚ__classdict__s   "€Új/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/distributed/optim/functional_adamax.pyÚ__annotate__Ú_FunctionalAdamax.__annotate__   sh   ø€ ÷ '.ñ '.á‘V•ð'.ñ ð'.ñ ‘U™E�\Õ"ð	'.ñ
 ð'.ñ ð'.ñ ð'.ñ ð'.ñ "&ñ'.ó    c	           
     ó¼  € \        ^R7       RV8:  g   \        RV 24      hRV8:  g   \        RV 24      hRV^ ,          u;8:  d   R8  g   M \        RV^ ,           24      hRV^,          u;8:  d   R8  g   M \        RV^,           24      hRV8:  g   \        RV 24      hR	VR
VRV^ ,          RV^,          RV/V n        W`n        Wpn        \
        P                  P                  \        \
        P                  \        \        \
        P                  3,          3,          / 4      V n        \        V4      ^ 8X  d   V'       g   \        R4      hRV/V n        R# )r   )Ú
stacklevelç        zInvalid learning rate: zInvalid epsilon value: g      ð?z#Invalid beta parameter at index 0: z#Invalid beta parameter at index 1: zInvalid weight_decay value: r
   r   Úbeta1Úbeta2r   z%optimizer got an empty parameter listr	   N)r   Ú
ValueErrorÚdefaultsr   r   ÚtorchÚjitÚannotateÚdictr   ÚstrÚstateÚlenÚparam_group)	Úselfr	   r
   r   r   r   r   r   r   s	   &&&&&&&&&r   Ú__init__Ú_FunctionalAdamax.__init__   sB  € ô 	;ÀaÕHØ�bŒyÜÐ6°r°dÐ;Ó<Ð<Ø�cŒzÜÐ6°s°eÐ<Ó=Ð=Ø�e˜A•hÖ$ Ö$ÜÐBÀ5ÈÅ8À*ÐMÓNÐNØ�e˜A•hÖ$ Ö$ÜÐBÀ5ÈÅ8À*ÐMÓNÐNØ�lÔ"ÜÐ;¸L¸>ÐJÓKÐKð �"Ø�3Ø�U˜1•XØ�U˜1•XØ˜Lð
ˆŒð ŒØ ŒÜ—Y‘Y×'Ñ'¬¬U¯\©\¼4ÄÄUÇ\Á\Ð@QÕ;RÐ-RÕ(SÐUWÓXˆŒ
äˆv‹;˜!Ô×$;ÜÐDÓEÐEð % fÐ-ˆÖr   c                ó>   <€ V ^8„  d   QhRS[ S[R,          ,          /# )r   Ú	gradientsN)r   r   )r   r   s   "€r   r   r   A   s   ø€ ÷ 8ñ 8™d¡6¨D¥=Õ1ñ 8r   c                ó0  € V P                   R ,          p. p. p. p. p. p\        V4      \        V4      8w  d3   \        RR\        V4       R2,           R\        V4       2,           4      hRp\        V P                   R ,          V4       EFC  w  ršV
f   K  V\        P
                  ! V	4      ,          pVP                  V	4       VP                  V
4       W�P                  9  dŽ   / V P                  V	&   V P                  V	,          p\        P                  ! R4      VR&   \        P                  ! V	\        P                  R	7      VR
&   \        P                  ! V	\        P                  R	7      VR&   V P                  V	,          pVP                  VR
,          4       VP                  VR,          4       VP                  VR,          4       EKF  	  \        P                  ! 4       ;_uu_ 4        \        P                  ! VVVVVV P                  R,          V P                  R,          V P                  R,          V P                  R,          V P                  R,          V P                  V P                   VR7       RRR4       R#   + '       g   i     R# ; i)r	   zEthe gradients passed in does not equal to the size of the parameters!zParams length: z. zGradients length: FNr   Ústep)Úmemory_formatÚexp_avgÚexp_infr   r   r   r
   r   )r   r   r   r
   r   r   r   Úhas_complex)r)   r(   r    Úzipr"   Ú
is_complexÚappendr'   ÚtensorÚ
zeros_likeÚpreserve_formatÚno_gradÚFÚadamaxr!   r   r   )r*   r.   r	   Úparams_with_gradÚgradsÚexp_avgsÚexp_infsÚstate_stepsr4   ÚparamÚgradientr'   s   &&          r   r0   Ú_FunctionalAdamax.stepA   s  € Ø×!Ñ! (Õ+ˆØÐØˆØˆØˆØ$&ˆäˆv‹;œ#˜i›.Ô(ÜØWØ#¤C¨£K =°Ð3õ4à&¤s¨9£~Ð&6Ð7õ8óð ð ˆÜ" 4×#3Ñ#3°HÕ#=¸y×I‰OˆEØÔ#Øœu×/Ò/°Ó6Õ6�Ø ×'Ñ'¨Ô.Ø—‘˜XÔ&à§
¡
Ô*Ø(*�D—J‘J˜uÑ%Ø ŸJ™J uÕ-�EÜ$)§L¢L°Ó$5�E˜&‘Mä',×'7Ò'7Ø¬U×-BÑ-Bô(�E˜)Ñ$ô (-×'7Ò'7Ø¬U×-BÑ-Bô(�E˜)Ñ$ð Ÿ
™
 5Õ)�à—‘  iÕ 0Ô1Ø—‘  iÕ 0Ô1Ø×"Ñ" 5¨¥=×1ñ/  Jô2 �]Š]�_�_Ü�HŠHØ ØØØØØ—M‘M %Õ(Ø—m‘m GÕ,Ø—m‘m GÕ,Ø—=‘= Õ&Ø!Ÿ]™]¨>Õ:ØŸ™ØŸ™Ø'õ÷ �_�_Š_ús   Ç-BJÊJ	)r!   r   r   r)   r'   N)gü©ñÒMbP?)gÍÌÌÌÌÌì?g+‡ÙÎ÷ï?g:Œ0âŽyE>r   FFF)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r+   r0   Ú__static_attributes__Ú__classdictcell__)r   s   @r   r   r      s   ø‡ € ÷'.ò '.÷R8ö 8r   r   c                óV   € V ^8„  d   Qh/ ^ \         9   d   \        \        ,          ;R&   # )r   Ú__all__)Ú__conditional_annotations__r   r&   )r   s   "r   r   r      s   € ‡�÷ Ò ŒŒc�Ñ ò r   )rN   r"   Útorch.optim._functionalÚoptimÚ_functionalr<   r   Ú,torch.distributed.optim._deprecation_warningr   rM   r   r   )rN   s   @r   Ú<module>rS      s1   øð÷ €ß #Ð #Ý õð
 €Ó ÷bó br   