PyYAML<8.0,>5.4
fsspec[http]<2027.0,>=2022.5.0
lightning-utilities<2.0,>=0.10.0
packaging<27.0,>=20.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.15.0,>=0.14.1

[data]
litdata<1.0,>=0.2.0rc

[dev]
cloudpickle<4.0,>=1.3
coverage==7.10.6
fastapi
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
numpy<2.0,>1.20.0
numpy<2.0,>=1.21.0
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==6.3.0
pytest-random-order==1.2.0
pytest-rerunfailures==16.0.1
pytest-timeout==2.4.0
pytest==8.4.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 != "Windows" and platform_system != "Darwin"]
deepspeed<=0.15.0,>=0.14.1

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

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

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

[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.15.0,>=0.14.1

[fabric-dev]
coverage==7.10.6
hydra-core<2.0,>=1.2.0
numpy<2.0,>=1.21.0
pytest-cov==6.3.0
pytest-random-order==1.2.0
pytest-rerunfailures==16.0.1
pytest-timeout==2.4.0
pytest==8.4.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.15.0,>=0.14.1

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

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

[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.15.0,>=0.14.1

[fabric-test]
coverage==7.10.6
numpy<2.0,>=1.21.0
pytest-cov==6.3.0
pytest-random-order==1.2.0
pytest-rerunfailures==16.0.1
pytest-timeout==2.4.0
pytest==8.4.2
tensorboardX<3.0,>=2.6

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

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

[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.15.0,>=0.14.1

[pytorch-dev]
cloudpickle<4.0,>=1.3
coverage==7.10.6
fastapi
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
numpy<2.0,>1.20.0
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==6.3.0
pytest-random-order==1.2.0
pytest-rerunfailures==16.0.1
pytest-timeout==2.4.0
pytest==8.4.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 != "Windows" and platform_system != "Darwin"]
deepspeed<=0.15.0,>=0.14.1

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

[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.15.0,>=0.14.1

[pytorch-test]
cloudpickle<4.0,>=1.3
coverage==7.10.6
fastapi
numpy<2.0,>1.20.0
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==6.3.0
pytest-random-order==1.2.0
pytest-rerunfailures==16.0.1
pytest-timeout==2.4.0
pytest==8.4.2
scikit-learn<2.0,>0.22.1
tensorboard<3.0,>=2.11
uvicorn

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

[strategies]

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

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

[test]
cloudpickle<4.0,>=1.3
coverage==7.10.6
fastapi
numpy<2.0,>1.20.0
numpy<2.0,>=1.21.0
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==6.3.0
pytest-random-order==1.2.0
pytest-rerunfailures==16.0.1
pytest-timeout==2.4.0
pytest==8.4.2
scikit-learn<2.0,>0.22.1
tensorboard<3.0,>=2.11
tensorboardX<3.0,>=2.6
uvicorn

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

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

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