Ross Wightman 13e0f3a4a3 Add loss scale arg, initial distributed loss scale. Maybe fix FX for the model.		2025-04-08 20:47:57 -07:00
..
data	Add loss scale arg, initial distributed loss scale. Maybe fix FX for the model.	2025-04-08 20:47:57 -07:00
layers	Initial NaFlex ViT model and training support	2025-04-07 21:27:10 -07:00
loss	…
models	Add loss scale arg, initial distributed loss scale. Maybe fix FX for the model.	2025-04-08 20:47:57 -07:00
optim	Fix issue taking device from V before V exists	2025-01-31 12:52:47 -08:00
scheduler	…
utils	…
__init__.py	…
py.typed	…
version.py	Update version.py	2025-02-22 10:50:21 -08:00