Worked on stabilizing the build system for the ROCm/rocm-jax repository, focusing on dependency management and packaging reliability. Addressed a recurring build-time error by ensuring the wheel package is explicitly installed during the build process, which previously led to failures when only auditwheel was managed. This targeted fix improved the robustness of continuous integration workflows and reduced friction for downstream users iterating on the package. Leveraged Python and build system expertise to enhance the packaging pipeline, resulting in more predictable and maintainable builds. The work emphasized careful dependency handling and contributed to a smoother development experience for the project’s users.