/Threepio

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

Primary LanguagePythonOtherNOASSERTION

Threepio logo

Coverage License OpenCollective

Threepio

Threepio makes it dead simple to translate commands between machine learning frameworks such as PyTorch, Tensorflow.js, and Tensorflow Python. It acts as a core component in PySyft and syft.js. Some of the life changing features this library provides are:

  • 🤖 Automatic bi-directional translation between same named commands add, abs, etc.
  • 🧠 Smart argument mapping between automaticaly translated commands
  • 🔁 Word level translation between libraries
  • 🔧 Custom command translation
  • 🌿 One to many command translation
  • ⚡ Translation of tensor methods (in progress)
  • 🔍 Support for fuzzy match in automatic command translation (in progress)
  • 🗂️ Support for framework versioning in automatic translation (in progress)
  • 🧙‍♂️ Support for automatic argument type casting (tensorflow tensor -> pytorch tensor) (in progress)

Threepio is made up of two main components:

And has clients in both Python and Javascript! And has a client available in Python!

Installation

Depending on your language, installation will differ.

Python

Use the package manager pip to install threepio.

pip install 3p0

Usage

Please check out our usage document for further guidelines.

Javascript

** Threepio.js has been deprecated. For existing javascript code, please refer to commit 0a538a3f **

Usage

Please check out our usage document for further guidelines.

Support

For support in using this library, please join the #lib_threepio Slack channel. If you'd like to follow along with any code changes to the library, please join the #code_threepio Slack channel. Click here to join our Slack community!

Contributing

Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.

Please make sure to update tests as appropriate.

Contributors

Thanks to these awesome people:


Mike Nolan avatar
Mike Nolan

OM FL Workers Team / Author
Amr Kayid avatar
Amr Kayid

Contributor
Vova Manannikov avatar
Vova Manannikov

OM FL Workers Team
Patrick Cason avatar
Patrick Cason

OM FL Workers Team Lead
Tvictor avatar
Tvictor

OM FL Workers Team
Raemsht Shukla avatar
Ramesht Shukla

OM FL Workers Team
Justin D. Harris avatar
Justin D. Harris

Contributor
Pedro Espíndula avatar
Pedro Espíndula

Contributor
Anirban Aaha avatar
Anirban Aaha

Contributor

License

Apache License 2.0