Ruinong Tian

Issue: keras tests are failing when running it for GPU . More background: We hard code keras to a legacy commit: https://github.com/aws/sagemaker-distribution/blob/main/test/test_artifacts/v1/keras.test.Dockerfile#L19 due to TensorFlow version conflict issue, but with...

*Issue #, if available:* *Description of changes:* 1. Make each major version has its own cuda and base image version 2. Update image key dependencies accordingly (including `langchain-aws` which was...

[BUG] Pytorch example failure in GPU environment

1

### Issues Policy acknowledgement - [X] I have read and agree to submit bug reports in accordance with the [issues policy](https://www.github.com/mlflow/mlflow/blob/master/ISSUE_POLICY.md) ### Where did you encounter this bug? Other ###...

s3fs is one of our marquee dependency introduced in 2.3.0, its unit test keep failing due to error: ``` s3 = def test_info(s3): s3.touch(a) s3.touch(b) info = s3.info(a) linfo =...