deepjavalibrary / djl-serving

A universal scalable machine learning model deployment solution
Apache License 2.0
183 stars 58 forks source link

[Serving] Implement SageMaker Secure Mode & support for multiple data sources #2042

Closed ethnzhng closed 1 month ago

ethnzhng commented 1 month ago

This draft PR adds the initial implementation of SageMaker Secure Mode, as well as support for multiple data sources.

I have tested the security control scenarios locally in Docker, and am currently working on adding unit tests and integration tests.

Summary of functionality

Basic support for additional model data sources

Specify trusted and untrusted data source paths

Configure individual DLC-level security controls which scan untrusted paths

Example scenarios: