autoray (conda): Abstract your array operations.
Keywords: array|dask|deep-learning|jax|lazy|machine-learning|numpy|python|pytorch|tensor|tensorflow