ben-walczak

Came here to say I'm getting the same error. I'm seeing everything listed correctly in the github source, but the pip install does not provide this as well

I'm not entirely sure. Just spitballing but I think it would be implemented somewhere within these lines of codes: https://github.com/luyug/GradCache/blob/main/src/grad_cache/grad_cache.py#L193C9-L211 and use similar logic to the following: ``` def gradfilter_ema(...