search

dgasmith / opt_einsum

⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
https://dgasmith.github.io/opt_einsum/
MIT License
863 stars 68 forks source link

Add Aesara backend #214

Closed rlouf closed 6 months ago

rlouf commented 1 year ago

Description

This PR adds an Aesara backend to opt_einsum following the discussion in #213.

Todos

Status

codecov[bot] commented 1 year ago

Codecov Report

Merging #214 (b1139d0) into master (1a984b7) will decrease coverage by 0.96%. The diff coverage is 33.33%.

jcmgray commented 1 year ago

All looks good to me, looking forward to testing the performance some time!

By the way @rlouf, I have an implementation of pairwise einsum that should be compatible with with aesara here - https://github.com/jcmgray/einsum_bmm, if aesara needed an implementation. (@dgasmith its also faster than numpy.einsum in many cases, might be something worth including here too....)

rlouf commented 1 year ago

That looks awesome! We have an issue that's been open for a while to implement einsum, it would be great to have it in the library.

dgasmith commented 1 year ago

@rlouf Looks like Asera tests are failing.

dgasmith commented 1 year ago

@rlouf Gentle ping here.

dgasmith commented 6 months ago

@rlouf Another gentle ping here, we will be shipping another release within the next week.

rlouf commented 6 months ago

I am sorry but I am not working on Aesara anymore. I am closing the PR for now.