rsokl / MyGrad

Drop-in autodiff for NumPy.
https://rsokl.github.io/MyGrad/
MIT License
195 stars 21 forks source link

Implement public interface for ufuncs #358

Closed rsokl closed 3 years ago

rsokl commented 3 years ago

mg.add, mg.sin, etc. are all implemented as mygrad-ufuncs. These are all registered subclasses of an un-inheritable abstract base class mg.ufunc.

Fixes a bug where mem-lock was not behaving correctly in conjunction with view targets, and when a boolean mask is specified.

Also adds matplotlib plots in some of the examples sections of various functions