Build backend, runtime, and compiler integrations with JAX, XLA, Pallas, MLIR, and related tooling; inspect compiler artifacts; work on lowering, fusion, and code generation; and benchmark production-relevant serving across correctness, latency, and throughput. Implement ideas from research papers; support mixture-of-experts, multimodal, diffusion, and agentic workloads; and improve scheduling, continuous batching, KV-cache memory management, prefix caching, and hybrid model serving.