Rocm dockerfile and test updates
Dockerfile updates Test script updates
@hawkinsp Please review. Thanks!
@hawkinsp @jakevdp @yashk2810 Any feedback on this? Can we please merge this? Thanks!
@skye @hawkinsp Can we please merge this? Thanks!
Please fix the lint failure.
@hawkinsp Should be fixed now. Please review. Thanks!
@hawkinsp @skye Can we please merge this? Thanks!
New lint failure that needs fixing.
Done as part of this https://github.com/google/jax/pull/22600