openmined.threepio 0.0.1

A multi-language library for translating commands between PyTorch, TensorFlow, and TensorFlow.js

Threepio Python Client

Threepio is a multi-language library for translating commands between PyTorch, TensorFlow, and TensorFlow.js This section of the code-base is for the Python Client.

Installation

Use the package manager pip to install threepio.

pip install openmined.threepio

Usage

import tensorflow as tf
from pythreepio import Threepio, Command

# Create a threepio translating from pytorch -> tensorflow python
threepio = Threepio("torch", "tf", tf)

# Create your command to translate
args = [
    tf.constant([[1, 0], [0, 1]]),
    tf.constant([[0, 1], [1, 0]])
]
kwargs = {}
cmd = Command("add", args, kwargs)

# Translate the command
translated_cmd = threepio.translate(cmd, lookup_command=True)

...

# When you're ready, execute the translation
translated_cmd.execute_routine() # -> [[1, 1], [1, 1]]

Command Support

Support for commands is currently limited. Officially supported commands are listed in our tests.