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
848 stars 68 forks source link

Finalizes typing and updates documentation #232

Closed dgasmith closed 4 months ago

dgasmith commented 4 months ago

Description

The PR provides a comprehensive typing overhaul which should provide a fully consistent typing paradigm. Given we will remove numpy as a dependency shortly (#204) we have several constraints on the typing update to avoid the np.typing module. While we can implement ArrayType as a Protocol, I'm unsure it's worth the effort.

Todos

Notable points that this PR has either accomplished or will accomplish.

Status

dgasmith commented 4 months ago

@jcmgray Final tweaks are in, please re-review! I'll tackle #204 next.