mithril-security / blindai

Confidential AI deployment with secure enclaves :lock:
https://www.mithrilsecurity.io/
Apache License 2.0
500 stars 35 forks source link

client: Accept numpy/torch tensors #80

Closed cchudant closed 2 years ago

cchudant commented 2 years ago

Description

Python client should accept numpy / torch tensors directly.

Motivation and Context

This would be a much better API.

Test plans

add unit tests

Additional Information

We should do it in a way that does not require users to install torch nor numpy, if they are not using this feature.

Checklist