EnzymeAD / EnzymeAD/Reactant.jl

Support sparse arrays (SparseArrays, Yao)

Open
#308 6 comments 0 reactions 0 assignees View on GitHub
Dominant language
Julia
Stars
370
Forks
74
Avg merge
18h 47m
Merged PRs (30d)
30

Description

MLIR has builtin support for sparse tensors and as far as we know, XLA supports them. We just haven't yet added support for nor tried them.

One case of particular interest for me is that Yao uses sparse arrays for representing some quantum gates. Specifically it uses a `LuxurySparseArray` type which has been giving me problems. Although it should be able to track the `TracedRNumber`s and end up with an `SparseArray{TracedRNumber}` kind of array, this is not what is happening (I'm running into errors many times) and it's also far from optimal (it's better to use a `TracedRArray`).

Currently, we force the dense representation of parametric Yao gates but we should reconsider the implementation and test it again once we add support for sparse tensors.

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.