Worked on upgrading the ROCm/rocm-jax repository to support ROCm 7.1.0 and JAX 0.7.1, focusing on improving compatibility and deployment readiness for downstream users. Updated CI/CD pipelines, Dockerfiles, and build scripts using Python, Shell, and YAML to ensure seamless integration with the latest ROCm and JAX releases. The technical approach emphasized robust build systems and Python packaging practices, validating changes through automated workflows. This work reduced build failures and integration risks, enabling faster validation of models within the ROCm ecosystem. No major bugs were addressed, as the primary goal was to align infrastructure with new software versions and requirements.