PyYAML<8.0,>5.4
fsspec[http]<2028.0,>=2022.5.0
lightning-utilities<2.0,>=0.10.0
packaging<27.0,>=23.0
torch<4.0,>=2.1.0
torchmetrics<3.0,>0.7.0
tqdm<6.0,>=4.57.0
typing-extensions<6.0,>4.5.0
pytorch-lightning

[all]
hydra-core<2.0,>=1.2.0
ipython[all]<11.0,>=8.0.0
jsonargparse[jsonnet,signatures]<5.0,>=4.39.0
matplotlib<4.0,>3.1
omegaconf<3.0,>=2.2.3
requests<3.0
rich<15.0,>=12.3.0
tensorboardX<3.0,>=2.2
torchmetrics<2.0,>=0.10.0
torchvision<1.0,>=0.16.0

[all:platform_system != "Darwin"]
bitsandbytes<1.0,>=0.45.2

[all:platform_system != "Windows" and platform_system != "Darwin"]
deepspeed<0.17.0,>=0.15.0

[data]
litdata<1.0,>=0.2.0rc

[dev]
cloudpickle<4.0,>=1.3
fastapi
huggingface-hub
hydra-core<2.0,>=1.2.0
ipython[all]<11.0,>=8.0.0
jsonargparse[jsonnet,signatures]<5.0,>=4.39.0
matplotlib<4.0,>3.1
omegaconf<3.0,>=2.2.3
onnx<2.0,>1.12.0
onnxruntime<2.0,>=1.12.0
onnxscript<1.0,>=0.1.0
pandas<3.0,>2.0
psutil<8.0
pytest-cov==7.0.0
pytest-random-order==1.2.0
pytest-timeout==2.4.0
pytest==9.0.2
requests<3.0
rich<15.0,>=12.3.0
scikit-learn<2.0,>0.22.1
tensorboard<3.0,>=2.11
tensorboardX<3.0,>=2.2
tensorboardX<3.0,>=2.6
torchmetrics<2.0,>=0.10.0
torchvision<1.0,>=0.16.0
uvicorn

[dev:platform_system != "Darwin"]
bitsandbytes<1.0,>=0.45.2

[dev:platform_system != "Darwin" and python_version >= "3.12"]
torch-tensorrt

[dev:platform_system != "Windows" and platform_system != "Darwin"]
deepspeed<0.17.0,>=0.15.0

[dev:python_version < "3.10"]
coverage==7.10.7
pytest-rerunfailures==16.0.1

[dev:python_version < "3.11"]
click==8.1.8

[dev:python_version < "3.12"]
numpy<2.0,>1.21.0

[dev:python_version > "3.10"]
click==8.3.1

[dev:python_version >= "3.10"]
coverage==7.13.1
pytest-rerunfailures==16.1

[dev:python_version >= "3.12"]
numpy<3.0,>2.1.0

[examples]
ipython[all]<11.0,>=8.0.0
requests<3.0
torchmetrics<2.0,>=0.10.0
torchvision<1.0,>=0.16.0

[extra]
hydra-core<2.0,>=1.2.0
jsonargparse[jsonnet,signatures]<5.0,>=4.39.0
matplotlib<4.0,>3.1
omegaconf<3.0,>=2.2.3
rich<15.0,>=12.3.0
tensorboardX<3.0,>=2.2

[extra:platform_system != "Darwin"]
bitsandbytes<1.0,>=0.45.2

[fabric-all]
hydra-core<2.0,>=1.2.0
torchmetrics<2.0,>=0.10.0
torchvision<1.0,>=0.16.0

[fabric-all:platform_system != "Darwin"]
bitsandbytes<1.0,>=0.45.2

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

[fabric-dev]
huggingface-hub
hydra-core<2.0,>=1.2.0
pytest-cov==7.0.0
pytest-random-order==1.2.0
pytest-timeout==2.4.0
pytest==9.0.2
tensorboardX<3.0,>=2.6
torchmetrics<2.0,>=0.10.0
torchvision<1.0,>=0.16.0

[fabric-dev:platform_system != "Darwin"]
bitsandbytes<1.0,>=0.45.2

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

[fabric-dev:python_version < "3.10"]
coverage==7.10.7
pytest-rerunfailures==16.0.1

[fabric-dev:python_version < "3.11"]
click==8.1.8

[fabric-dev:python_version < "3.12"]
numpy<2.0,>1.21.0

[fabric-dev:python_version > "3.10"]
click==8.3.1

[fabric-dev:python_version >= "3.10"]
coverage==7.13.1
pytest-rerunfailures==16.1

[fabric-dev:python_version >= "3.12"]
numpy<3.0,>2.1.0

[fabric-examples]
torchmetrics<2.0,>=0.10.0
torchvision<1.0,>=0.16.0

[fabric-extra]
hydra-core<2.0,>=1.2.0

[fabric-strategies]

[fabric-strategies:platform_system != "Darwin"]
bitsandbytes<1.0,>=0.45.2

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

[fabric-test]
huggingface-hub
pytest-cov==7.0.0
pytest-random-order==1.2.0
pytest-timeout==2.4.0
pytest==9.0.2
tensorboardX<3.0,>=2.6

[fabric-test:python_version < "3.10"]
coverage==7.10.7
pytest-rerunfailures==16.0.1

[fabric-test:python_version < "3.11"]
click==8.1.8

[fabric-test:python_version < "3.12"]
numpy<2.0,>1.21.0

[fabric-test:python_version > "3.10"]
click==8.3.1

[fabric-test:python_version >= "3.10"]
coverage==7.13.1
pytest-rerunfailures==16.1

[fabric-test:python_version >= "3.12"]
numpy<3.0,>2.1.0

[pytorch-all]
hydra-core<2.0,>=1.2.0
ipython[all]<11.0,>=8.0.0
jsonargparse[jsonnet,signatures]<5.0,>=4.39.0
matplotlib<4.0,>3.1
omegaconf<3.0,>=2.2.3
requests<3.0
rich<15.0,>=12.3.0
tensorboardX<3.0,>=2.2
torchmetrics<2.0,>=0.10.0
torchvision<1.0,>=0.16.0

[pytorch-all:platform_system != "Darwin"]
bitsandbytes<1.0,>=0.45.2

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

[pytorch-dev]
cloudpickle<4.0,>=1.3
fastapi
huggingface-hub
hydra-core<2.0,>=1.2.0
ipython[all]<11.0,>=8.0.0
jsonargparse[jsonnet,signatures]<5.0,>=4.39.0
matplotlib<4.0,>3.1
omegaconf<3.0,>=2.2.3
onnx<2.0,>1.12.0
onnxruntime<2.0,>=1.12.0
onnxscript<1.0,>=0.1.0
pandas<3.0,>2.0
psutil<8.0
pytest-cov==7.0.0
pytest-random-order==1.2.0
pytest-timeout==2.4.0
pytest==9.0.2
requests<3.0
rich<15.0,>=12.3.0
scikit-learn<2.0,>0.22.1
tensorboard<3.0,>=2.11
tensorboardX<3.0,>=2.2
torchmetrics<2.0,>=0.10.0
torchvision<1.0,>=0.16.0
uvicorn

[pytorch-dev:platform_system != "Darwin"]
bitsandbytes<1.0,>=0.45.2

[pytorch-dev:platform_system != "Darwin" and python_version >= "3.12"]
torch-tensorrt

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

[pytorch-dev:python_version < "3.10"]
coverage==7.10.7
pytest-rerunfailures==16.0.1

[pytorch-dev:python_version < "3.12"]
numpy<2.0,>1.21.0

[pytorch-dev:python_version >= "3.10"]
coverage==7.13.1
pytest-rerunfailures==16.1

[pytorch-dev:python_version >= "3.12"]
numpy<3.0,>2.1.0

[pytorch-examples]
ipython[all]<11.0,>=8.0.0
requests<3.0
torchmetrics<2.0,>=0.10.0
torchvision<1.0,>=0.16.0

[pytorch-extra]
hydra-core<2.0,>=1.2.0
jsonargparse[jsonnet,signatures]<5.0,>=4.39.0
matplotlib<4.0,>3.1
omegaconf<3.0,>=2.2.3
rich<15.0,>=12.3.0
tensorboardX<3.0,>=2.2

[pytorch-extra:platform_system != "Darwin"]
bitsandbytes<1.0,>=0.45.2

[pytorch-strategies]

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

[pytorch-test]
cloudpickle<4.0,>=1.3
fastapi
huggingface-hub
onnx<2.0,>1.12.0
onnxruntime<2.0,>=1.12.0
onnxscript<1.0,>=0.1.0
pandas<3.0,>2.0
psutil<8.0
pytest-cov==7.0.0
pytest-random-order==1.2.0
pytest-timeout==2.4.0
pytest==9.0.2
scikit-learn<2.0,>0.22.1
tensorboard<3.0,>=2.11
uvicorn

[pytorch-test:platform_system != "Darwin" and python_version >= "3.12"]
torch-tensorrt

[pytorch-test:python_version < "3.10"]
coverage==7.10.7
pytest-rerunfailures==16.0.1

[pytorch-test:python_version < "3.12"]
numpy<2.0,>1.21.0

[pytorch-test:python_version >= "3.10"]
coverage==7.13.1
pytest-rerunfailures==16.1

[pytorch-test:python_version >= "3.12"]
numpy<3.0,>2.1.0

[strategies]

[strategies:platform_system != "Darwin"]
bitsandbytes<1.0,>=0.45.2

[strategies:platform_system != "Windows" and platform_system != "Darwin"]
deepspeed<0.17.0,>=0.15.0

[test]
cloudpickle<4.0,>=1.3
fastapi
huggingface-hub
onnx<2.0,>1.12.0
onnxruntime<2.0,>=1.12.0
onnxscript<1.0,>=0.1.0
pandas<3.0,>2.0
psutil<8.0
pytest-cov==7.0.0
pytest-random-order==1.2.0
pytest-timeout==2.4.0
pytest==9.0.2
scikit-learn<2.0,>0.22.1
tensorboard<3.0,>=2.11
tensorboardX<3.0,>=2.6
uvicorn

[test:platform_system != "Darwin" and python_version >= "3.12"]
torch-tensorrt

[test:python_version < "3.10"]
coverage==7.10.7
pytest-rerunfailures==16.0.1

[test:python_version < "3.11"]
click==8.1.8

[test:python_version < "3.12"]
numpy<2.0,>1.21.0

[test:python_version > "3.10"]
click==8.3.1

[test:python_version >= "3.10"]
coverage==7.13.1
pytest-rerunfailures==16.1

[test:python_version >= "3.12"]
numpy<3.0,>2.1.0
