search

Jutho / TensorOperations.jl

Julia package for tensor contractions and related operations
https://jutho.github.io/TensorOperations.jl/stable/
Other
453 stars 56 forks source link

gpu support #61

Closed jumerckx closed 5 years ago

jumerckx commented 5 years ago

Hello, I love the performance and ease of use this package offers, however I was interested in using TensorOperations in a Flux model but gpu is currently not supported Would it be possible to support CuArrays with TensorOperations?

Many thanks Jules

pgawron commented 5 years ago

Definitively it is a feature that would be also very useful for QuantumInformation.jl.

Jutho commented 5 years ago

Going live: https://discourse.julialang.org/t/pre-ann-cutensoroperations-jl/27305 and https://github.com/Jutho/CuTensorOperations.jl