There's an effort to have a drop in replacement for numpy using arrayfire. https://github.com/FilipeMaia/afnumpy

It is still a work in progress and requires some upstream changes in arrayfire to support the numpy api better.