tf-encrypted / moose

Secure distributed dataflow framework for encrypted machine learning and data processing
Apache License 2.0
57 stars 15 forks source link

Add `GrpcRuntime` to PyMoose #1083

Closed mortendahl closed 2 years ago

mortendahl commented 2 years ago

This PR adds gRPC runtime bindings to PyMoose:

runtime = pm.GrpcMooseRuntime(
  role_assignment={
    "alice": "localhost:50000",
    "bob": "localhost:50001",
    "carole": "localhost:50002",
  }
)

results = runtime.evaluate_computation(my_computation)

Follow-up: https://github.com/tf-encrypted/moose/issues/1090 https://github.com/tf-encrypted/moose/issues/1089 https://github.com/tf-encrypted/moose/issues/1094