[Feat] Added JAX-Triton bridge for ROCm#649

AllenFarcas wants to merge 5 commits into

devROCm/TransformerEngine:devfrom

alfarcas/jax-triton-bridgeROCm/TransformerEngine:alfarcas/jax-triton-bridgeCopy head branch name to clipboard