Ross Wightman cd21e80d03 Fiddling with Kron (PSGD)

2025-01-27 21:02:26 -08:00

..

data

…

layers

Add missing training flag to convert_sync_batchnorm

2025-01-21 11:51:55 -08:00

loss

…

models

Fix RDNet forward call

2025-01-21 11:52:05 -08:00

optim

Fiddling with Kron (PSGD)

2025-01-27 21:02:26 -08:00

scheduler

…

utils

…

__init__.py

Use import alias

2025-01-22 10:27:17 -08:00

py.typed

…

version.py

Version 1.0.14, update README & changelog

2025-01-19 13:53:09 -08:00