arrayfire python-pytorch root tensorflow