jax

d2ca35a7 - Use specific exception types in get_rocm_shared_memory_limit

Use specific exception types in get_rocm_shared_memory_limit Address review feedback: catch only expected exceptions from subprocess.run (FileNotFoundError, TimeoutExpired, OSError) instead of bare Exception.