jax
4dd54a8a - [ROCm]: Add Jax-Triton support for ROCm

Commit

2 years ago

[ROCm]: Add Jax-Triton support for ROCm

References

rocm-jaxlib-v0.4.12-jax-triton-rocm55

Author

Rahul Batra

Committer

Rahul Batra

Parents

Loading