chanfig>=0.0.109
gitpython
lazy-imports
torchdata
tqdm

[:python_version < "3.11"]
strenum

[accelerate]
accelerate
torch
torchdata
torcheval
torchmetrics

[deepspeed]
deepspeed
torch
torchdata
torcheval
torchmetrics

[jax]
flax
jax

[tensorflow]
tensorflow

[torch]
torch
torchdata
torchmetrics
