Worked on the ROCm/jax repository to deliver a focused documentation update for the JAX Callback API, clarifying that callback functions must not invoke JAX within pure_callback, io_callback, or debug_callback. This change addressed a key area of developer confusion by aligning documentation with the API contract, thereby reducing the risk of misuse and streamlining onboarding for new contributors. The work emphasized API design principles and careful documentation practices, utilizing Markdown and Python to ensure clarity and accuracy. No code changes or bug fixes were required, as the primary goal was to strengthen guidance and support efficiency for the developer community.