Ross Wightman b4bb0f452a Exclude embeds module and mask attn functions from tracing

2025-04-09 15:34:15 -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

Exclude embeds module and mask attn functions from tracing

2025-04-09 15:34:15 -07:00

optim

Fix issue taking device from V before V exists

2025-01-31 12:52:47 -08:00

scheduler

…

utils

Fix typos

2025-01-19 13:39:40 -08:00

__init__.py

Use import alias

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

py.typed

…

version.py

Update version.py

2025-02-22 10:50:21 -08:00