Official weights for Celo2-base learned update rule proposed in paper: Celo2: Towards Learned Optimization Free Lunch
Code repo: https://github.com/amoudgl/celo2