Worked on the jax-ml/jax repository to enhance the correctness and stability of VMapped computations by addressing a critical bug related to mutable array aliasing. Developed and implemented a fix in Python that disallows passing the same mutable array multiple times to a vmapped function when mutable array checks are enabled, thereby preventing incorrect computations and improving determinism in VMAP paths. Focused on robust debugging and thorough testing to ensure the patch was reliable and auditable. This contribution increased the reliability of vmap for users working with mutable arrays, demonstrating strong skills in Python, data science, and machine learning workflows.