Modify GPT_J to use easydel

[ctr-pmuruganTT]

Problem description

Add EasyDel implementation for JAX Whisper model

What's changed

• updated requirements.txt with the requirements for this model
• updated load_model() to use EasyDel model
• Added gradient_checkpointing=EasyDeLGradientCheckPointers.NONE as default it is taking a bool value and it is resulting in an error as ValueError: Invalid policy type: <class 'bool'>
• updated load_inputs() to pass mesh as a parameter
• added load_parameters_partition_spec() and get_input_activations_partition_spec()

Checklist

• New/Existing tests provide coverage for changes

• Tested in local with these changes without gradient_checkpointing - gpt_j_debug.log

• Tested by adding gradient_checkpointing - gpt_j_update_config_kwargs.log