Worked on stabilizing the build process for the apple/axlearn repository by addressing a compatibility issue with NVIDIA NCCL and CUDA12. Focused on dependency management, the developer pinned the nvidia-nccl-cu12 package to version 2.27.5, preventing the jax[cuda12] package from introducing newer, incompatible NCCL versions. This approach improved build reproducibility and reduced CI flakiness, ensuring consistent environments across development and testing. The work demonstrated disciplined package management and provided clear commit-level traceability for future audits. Python was used throughout, with an emphasis on robust dependency handling to maintain compatibility and reliability in the continuous integration workflow.