search

BaguaSys / bagua

Bagua Speeds up PyTorch
https://tutorials-8ro.pages.dev/
MIT License
875 stars 83 forks source link

Does it support Tensorflow? #153

Closed BobLiu20 closed 3 years ago

BobLiu20 commented 3 years ago

I see example only with PyTorch

NOBLES5E commented 3 years ago

Currently the bagua python package does not support tensorflow. There are some undergoing work happening in bagua-core so that tensorflow can use its C API for distributed learning, but it is created for internal scenarios and nowhere near complete.

NOBLES5E commented 3 years ago

Feel free to let me know if there are any other questions :)