search

ACEsuit / Polynomials4ML.jl

Polynomials for ML: fast evaluation, batching, differentiation
MIT License
12 stars 5 forks source link

CachedArrays for Linear layer #56

Closed CheukHinHoJerry closed 1 year ago

CheukHinHoJerry commented 1 year ago

Current we do not make good use of CachedArray in LinearLayer. Should implement this and test it someday.

CheukHinHoJerry commented 1 year ago

I am thinking to implement this but I am uncertain whether the CachedArray should go into st or the layer itself. @cortner Would be great if you can advice.

cortner commented 1 year ago

Both. And depending on use case one or the other should be called. (I think!)