Developed foundational infrastructure for the ROCm/jax repository to enable OneAPI Intel GPU plugin extension, laying the groundwork for future Intel GPU support within JAX. The work involved implementing C++ glue code and Python build configurations to bridge OneAPI with existing JAX functionalities, facilitating seamless cross-backend integration. By introducing integration hooks and plugin extension mechanisms, the developer expanded the hardware compatibility of JAX, allowing for hardware-accelerated workloads on Intel GPUs. This effort leveraged skills in C++ development, GPU programming, and plugin development, ultimately broadening the OneAPI ecosystem and accelerating time-to-market for cross-vendor acceleration within the JAX framework.