jax

9fc59609 - [0.6.0-UT] Adding third field to return value of get_rocm_version (#542)