Rather than use multiple ML frameworks, we want to simplify and use either torch or tensorflow. I think torch is the better option because of PyTorch's dynamic computation graph, ease of debugging, and its increasing popularity in the research community.
Once refactored, please test and make sure the demos are working with similar results. And then remove tensorflow packages from pyproject.toml and installation from Docker.
Tensorflow is used in:
Rather than use multiple ML frameworks, we want to simplify and use either torch or tensorflow. I think torch is the better option because of PyTorch's dynamic computation graph, ease of debugging, and its increasing popularity in the research community.
Once refactored, please test and make sure the demos are working with similar results. And then remove tensorflow packages from pyproject.toml and installation from Docker.
Rebuild and publish Docker images accordingly.