search

xnd-project / libxnd

Subsumed into xnd
https://xnd.io/
BSD 3-Clause "New" or "Revised" License
81 stars 12 forks source link

Try out integration with Tensor Computations #9

Open saulshanabrook opened 6 years ago

saulshanabrook commented 6 years ago

The Tensor Computations library provides a succinct and efficient way of writing computations on arrays. It also agnostic to the memory backend. I think we should be able to provide an xnd wrapper, so that memory is allocated in xnd, but can be computed with TC. See https://facebookresearch.github.io/TensorComprehensions/integrating_any_ml_framework.html

We have to be able to attach some metadata to xnd tensors to be able to wrap them in the dlpack format: https://github.com/dmlc/dlpack

saulshanabrook commented 6 years ago

Possibly relevant issue around nesting function computation https://github.com/facebookresearch/TensorComprehensions/issues/56