PyYAML<8.0,>=5.4
fsspec[http]<2026.0,>=2022.5.0
lightning-utilities<2.0,>=0.10.0
numpy<3.0,>=1.17.2
packaging<25.0,>=20.0
torch<4.0,>=2.0.0
torchmetrics<3.0,>=0.7.0
tqdm<6.0,>=4.57.0
typing-extensions<6.0,>=4.4.0
pytorch-lightning

[all]
bitsandbytes<1.0,>=0.42.0
hydra-core<2.0,>=1.2.0
ipython[all]<9.0
jsonargparse[signatures]<5.0,>=4.27.7
lightning-utilities<1.0,>=0.8.0
matplotlib<4.0,>3.1
omegaconf<3.0,>=2.2.3
requests<3.0
rich<14.0,>=12.3.0
tensorboardX<3.0,>=2.2
torchmetrics<2.0,>=0.10.0
torchvision<1.0,>=0.15.0

[all:platform_system != "Windows" and platform_system != "Darwin"]
deepspeed<=0.9.3,>=0.8.2

[app]
lightning_app<3.0,>=2.3.3

[data]
litdata<1.0,>=0.2.0rc

[dev]
bitsandbytes<1.0,>=0.42.0
click==8.1.7
cloudpickle<3.0,>=1.3
coverage==7.3.1
fastapi
hydra-core<2.0,>=1.2.0
ipython[all]<9.0
jsonargparse[signatures]<5.0,>=4.27.7
lightning-utilities<1.0,>=0.8.0
matplotlib<4.0,>3.1
omegaconf<3.0,>=2.2.3
onnx<2.0,>=0.14.0
onnxruntime<2.0,>=0.15.0
pandas<3.0,>1.0
psutil<6.0
pytest-cov==4.1.0
pytest-random-order==1.1.0
pytest-rerunfailures==12.0
pytest-timeout==2.1.0
pytest==7.4.0
requests<3.0
rich<14.0,>=12.3.0
scikit-learn<2.0,>0.22.1
tensorboard<3.0,>=2.9.1
tensorboardX<3.0,>=2.2
torchmetrics<2.0,>=0.10.0
torchmetrics<2.0,>=0.7.0
torchvision<1.0,>=0.15.0
uvicorn

[dev:platform_system != "Windows" and platform_system != "Darwin"]
deepspeed<=0.9.3,>=0.8.2

[examples]
ipython[all]<9.0
lightning-utilities<1.0,>=0.8.0
requests<3.0
torchmetrics<2.0,>=0.10.0
torchvision<1.0,>=0.15.0

[extra]
bitsandbytes<1.0,>=0.42.0
hydra-core<2.0,>=1.2.0
jsonargparse[signatures]<5.0,>=4.27.7
matplotlib<4.0,>3.1
omegaconf<3.0,>=2.2.3
rich<14.0,>=12.3.0
tensorboardX<3.0,>=2.2

[fabric-all]
bitsandbytes<1.0,>=0.42.0
lightning-utilities<1.0,>=0.8.0
torchmetrics<2.0,>=0.10.0
torchvision<1.0,>=0.15.0

[fabric-all:platform_system != "Windows" and platform_system != "Darwin"]
deepspeed<=0.9.3,>=0.8.2

[fabric-dev]
bitsandbytes<1.0,>=0.42.0
click==8.1.7
coverage==7.3.1
lightning-utilities<1.0,>=0.8.0
pytest-cov==4.1.0
pytest-random-order==1.1.0
pytest-rerunfailures==12.0
pytest-timeout==2.1.0
pytest==7.4.0
tensorboardX<3.0,>=2.2
torchmetrics<2.0,>=0.10.0
torchmetrics<2.0,>=0.7.0
torchvision<1.0,>=0.15.0

[fabric-dev:platform_system != "Windows" and platform_system != "Darwin"]
deepspeed<=0.9.3,>=0.8.2

[fabric-examples]
lightning-utilities<1.0,>=0.8.0
torchmetrics<2.0,>=0.10.0
torchvision<1.0,>=0.15.0

[fabric-strategies]
bitsandbytes<1.0,>=0.42.0

[fabric-strategies:platform_system != "Windows" and platform_system != "Darwin"]
deepspeed<=0.9.3,>=0.8.2

[fabric-test]
click==8.1.7
coverage==7.3.1
pytest-cov==4.1.0
pytest-random-order==1.1.0
pytest-rerunfailures==12.0
pytest-timeout==2.1.0
pytest==7.4.0
tensorboardX<3.0,>=2.2
torchmetrics<2.0,>=0.7.0

[pytorch-all]
bitsandbytes<1.0,>=0.42.0
hydra-core<2.0,>=1.2.0
ipython[all]<9.0
jsonargparse[signatures]<5.0,>=4.27.7
lightning-utilities<1.0,>=0.8.0
matplotlib<4.0,>3.1
omegaconf<3.0,>=2.2.3
requests<3.0
rich<14.0,>=12.3.0
tensorboardX<3.0,>=2.2
torchmetrics<2.0,>=0.10.0
torchvision<1.0,>=0.15.0

[pytorch-all:platform_system != "Windows" and platform_system != "Darwin"]
deepspeed<=0.9.3,>=0.8.2

[pytorch-dev]
bitsandbytes<1.0,>=0.42.0
cloudpickle<3.0,>=1.3
coverage==7.3.1
fastapi
hydra-core<2.0,>=1.2.0
ipython[all]<9.0
jsonargparse[signatures]<5.0,>=4.27.7
lightning-utilities<1.0,>=0.8.0
matplotlib<4.0,>3.1
omegaconf<3.0,>=2.2.3
onnx<2.0,>=0.14.0
onnxruntime<2.0,>=0.15.0
pandas<3.0,>1.0
psutil<6.0
pytest-cov==4.1.0
pytest-random-order==1.1.0
pytest-rerunfailures==12.0
pytest-timeout==2.1.0
pytest==7.4.0
requests<3.0
rich<14.0,>=12.3.0
scikit-learn<2.0,>0.22.1
tensorboard<3.0,>=2.9.1
tensorboardX<3.0,>=2.2
torchmetrics<2.0,>=0.10.0
torchvision<1.0,>=0.15.0
uvicorn

[pytorch-dev:platform_system != "Windows" and platform_system != "Darwin"]
deepspeed<=0.9.3,>=0.8.2

[pytorch-examples]
ipython[all]<9.0
lightning-utilities<1.0,>=0.8.0
requests<3.0
torchmetrics<2.0,>=0.10.0
torchvision<1.0,>=0.15.0

[pytorch-extra]
bitsandbytes<1.0,>=0.42.0
hydra-core<2.0,>=1.2.0
jsonargparse[signatures]<5.0,>=4.27.7
matplotlib<4.0,>3.1
omegaconf<3.0,>=2.2.3
rich<14.0,>=12.3.0
tensorboardX<3.0,>=2.2

[pytorch-strategies]

[pytorch-strategies:platform_system != "Windows" and platform_system != "Darwin"]
deepspeed<=0.9.3,>=0.8.2

[pytorch-test]
cloudpickle<3.0,>=1.3
coverage==7.3.1
fastapi
onnx<2.0,>=0.14.0
onnxruntime<2.0,>=0.15.0
pandas<3.0,>1.0
psutil<6.0
pytest-cov==4.1.0
pytest-random-order==1.1.0
pytest-rerunfailures==12.0
pytest-timeout==2.1.0
pytest==7.4.0
scikit-learn<2.0,>0.22.1
tensorboard<3.0,>=2.9.1
uvicorn

[store-test]
coverage==7.3.1
pytest-cov==4.1.0
pytest-random-order==1.1.0
pytest-rerunfailures==12.0
pytest-timeout==2.1.0
pytest==7.4.0

[strategies]
bitsandbytes<1.0,>=0.42.0

[strategies:platform_system != "Windows" and platform_system != "Darwin"]
deepspeed<=0.9.3,>=0.8.2

[test]
click==8.1.7
cloudpickle<3.0,>=1.3
coverage==7.3.1
fastapi
onnx<2.0,>=0.14.0
onnxruntime<2.0,>=0.15.0
pandas<3.0,>1.0
psutil<6.0
pytest-cov==4.1.0
pytest-random-order==1.1.0
pytest-rerunfailures==12.0
pytest-timeout==2.1.0
pytest==7.4.0
scikit-learn<2.0,>0.22.1
tensorboard<3.0,>=2.9.1
tensorboardX<3.0,>=2.2
torchmetrics<2.0,>=0.7.0
uvicorn
