Fiddling with Kron (PSGD) optimizer#2427

Merged

rwightman merged 5 commits intomainfrom

Jan 28, 2025

Commits on Jan 24, 2025

Fiddling with Kron (PSGD)
rwightman
committed

Commits on Jan 27, 2025

Some more kron work. Figured out why some tests fail, implemented a deterministic rng state load but too slow so skipping some tests for now.
rwightman
committed
More additions to Kron
rwightman
committed
Move opt_einsum import back out of class __init__
rwightman
committed

Commits on Jan 28, 2025

Prep Kron for merge, add detail to attributions note, README.
rwightman
committed