opt_einsum icon indicating copy to clipboard operation
opt_einsum copied to clipboard
verified publisher icon dgasmith

Add Aesara backend

Open rlouf opened this issue 2 years ago • 5 comments

Description

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

Todos

  • Add Aesara backend
  • Add relevant tests
  • Updated dev environment
  • Updated documentation and README

Status

  • [ ] Ready to go

rlouf avatar Mar 10 '23 09:03 rlouf

Codecov Report

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

codecov[bot] avatar Mar 10 '23 09:03 codecov[bot]

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....)

jcmgray avatar Mar 24 '23 20:03 jcmgray

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.

rlouf avatar Mar 26 '23 09:03 rlouf

@rlouf Looks like Asera tests are failing.

dgasmith avatar Apr 01 '23 16:04 dgasmith

@rlouf Gentle ping here.

dgasmith avatar May 26 '23 23:05 dgasmith

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

dgasmith avatar May 27 '24 14:05 dgasmith

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

rlouf avatar May 27 '24 16:05 rlouf