opt_einsum icon indicating copy to clipboard operation
opt_einsum copied to clipboard

Published 20 hours ago • verified publisher icon dgasmith

Support for Python array API standard

Open IsaacBreen opened this issue 2 years ago • 4 comments

Description

This PR adds support for libraries that conform to the Python array API standard. It discovers complying modules upon import of opt_einsum.

Eventually (and hopefully soon), once the array API standard becomes ubiquitous, this should largely eliminate the need for backend-specific code.

Note that to test this you'll need a version of an array library which implements the API. Right now AFAIK closest thing to this is latest (dev) Numpy which you can install using

pip install --upgrade git+https://github.com/numpy/numpy

Todos

  • [x] Tests
  • [x] Update docs

Status

  • [x] Ready to go

IsaacBreen avatar Jul 17 '22 07:07 IsaacBreen

Codecov Report

Merging #197 (13ec280) into master (3824e4f) will decrease coverage by 0.03%. The diff coverage is 94.59%.

codecov[bot] avatar Jul 18 '22 04:07 codecov[bot]

Looks like plenty of errors for Py3.6 where importlib.metadata is not yet extant.

dgasmith avatar Jul 18 '22 04:07 dgasmith

Thanks Daniel! The errors for Python <= 3.7 should be fixed. I've set it to bypass array API backend discovery altogether in such cases. Numpy doesn't support <= 3.7 as-is, so I don't think this will be missed.

I've fixed up the formatting too.

IsaacBreen avatar Jul 18 '22 09:07 IsaacBreen

@IsaacBreen I'm looking to push out another release in the next month or so. Is this a PR which you would like to support?

dgasmith avatar May 05 '24 18:05 dgasmith