






'''simple docstring'''







import jax.numpy as jnp

from ...utils import logging
from ..ta.modeling_flax_ta import FlaxTaEncoderModel, FlaxTaForConditionalGeneration, FlaxTaModel
from .configuration_mta import MTaConfig


UpperCamelCase    :      Optional[int]  			=		logging.get_logger(__name__)

UpperCamelCase    :      Dict  			=		'T5Config'
def 						A__			(				__lowerCAmelCase		:			jnp.array   ,    __lowerCAmelCase		:			int   ,    __lowerCAmelCase		:			int						):
						lowerCamelCase__         =  jnp.zeros_like(__lowerCAmelCase						)
						lowerCamelCase__         =  shifted_input_ids.at[:, 1:].set(input_ids[:, :-1]						)
						lowerCamelCase__         =  shifted_input_ids.at[:, 0].set(__lowerCAmelCase						)

						lowerCamelCase__         =  jnp.where(shifted_input_ids == -100   ,    __lowerCAmelCase   ,    __lowerCAmelCase						)
						return shifted_input_ids




class 		UpperCamelCase__    (a			):




		'''simple docstring'''





		_UpperCamelCase				  =					'mt5'
		_UpperCamelCase				  =					MTaConfig




class 		UpperCamelCase__    (a			):




		'''simple docstring'''





		_UpperCamelCase				  =					'mt5'
		_UpperCamelCase				  =					MTaConfig




class 		UpperCamelCase__    (a			):




		'''simple docstring'''





		_UpperCamelCase				  =					'mt5'
		_UpperCamelCase				  =					MTaConfig









'''simple docstring'''







def 						A__			(				):
						return [
						    a * b * (1000 - a - b)
						    for a in range(1   ,    999						)
						    for b in range(__lowerCAmelCase   ,    999						)
						    if (a * a + b * b == (1000 - a - b) ** 2)
						][0]


if __name__ == "__main__":
							print(F'{solution() = }')

