Support Advanced Profiler Options for AWS Neuron Trainium Hardware

[aws-shreya]

I want to request support for advanced profiler options for AWS Trainium Hardware. Currently JAX supports the Advanced Profiling Options mentioned here but is there any validation on which key value pairs can be passed as these advanced options or are there plans to add validation for these?
@sannidhyachauhan can you help with this as you previously added the advanced options.