Complete run_pytest_rocm.sh for CI integration
Add standard JAX CI components to run_pytest_rocm.sh: bash header with
license, environment setup (source default.env and install_wheels_locally.sh),
diagnostic output (pip freeze, device check, rocm-smi), and proper pytest
invocation using $JAXCI_PYTHON.
Set XLA flags: --xla_gpu_force_compilation_parallelism=1,
--xla_gpu_enable_nccl_comm_splitting=false, and --xla_gpu_enable_command_buffer=
Remove CSV/HTML reporting and log redirection for initial implementation,
marked with TODOs for future consideration.
