mmbejani/gvr-optimizer

A Pytorch & Jax implementation of gradient variance reducer optimization algorithm