Ross Wightman 2b251fb291 Wrap torch checkpoint() fn to default use_reentrant flag to False and allow env var override		2025-01-06 11:28:39 -08:00
..
data	Make sure trust_remote code only passed to HF datasets. Improve some docstrings.	2024-12-06 11:40:04 -08:00
layers	Wrap torch checkpoint() fn to default use_reentrant flag to False and allow env var override	2025-01-06 11:28:39 -08:00
loss	…
models	Wrap torch checkpoint() fn to default use_reentrant flag to False and allow env var override	2025-01-06 11:28:39 -08:00
optim	Add caution to Adan. Add decouple decay option to LAMB.	2024-12-05 13:50:30 -08:00
scheduler	…
utils	…
__init__.py	…
py.typed	…
version.py	Back to dev version	2024-12-06 10:44:41 -08:00