mirror of
https://github.com/ROCm/jax.git
synced 2025-04-21 06:16:06 +00:00
8552852609
This commit adds the scripts and envs that set up a Docker container named "jax". It is used in internal JAX CI jobs that handle building and publishing JAX artifacts to PyPI and/or GCS. While GitHub action workflows use the same Docker images, they do not run `run_docker_container.sh` script as they leverage built-in containerization features to run jobs within a container. PiperOrigin-RevId: 706793601