You might be interested in Legate [1]. It supports the NumPy interface as a drop-in replacement, supports GPUs and also distributed machines. And you can see fo

You might be interested in Legate [1]. It supports the NumPy interface as a drop-in replacement, supports GPUs and also distributed machines. And you can see for yourself their performance results; they're not far off from hand-tuned MPI. [1]: https://github.com/nv-legate/legate.numpy https://github.com/nv-legate/legate.numpy Disclaimer: I work on the library Legate uses for distributed computing, but otherwise have no connection.