requests
numpy
torch>=2.2.0
torchvision>=0.17.0
tqdm
transformers<6.0.0,>=5.1.0
pydantic<3,>=2.0
supervision>=0.29.0
pyDeprecate<0.10,>=0.9

[augment]
albumentations<3.0.0,>=1.4.24
kornia<1,>=0.7

[cli]
jsonargparse[signatures]>=4.27.7

[coreai]

[coreai:python_version >= "3.11" and python_version < "3.14" and ((sys_platform == "darwin" and platform_machine == "arm64") or (sys_platform == "linux" and platform_machine == "x86_64"))]
coreai-torch<0.5,>=0.4.2

[coreml]
coremltools<10.0,>=8.0

[cuda]

[cuda:sys_platform == "linux" and platform_machine == "x86_64"]
transformer-engine[pytorch]<3,>=2.19

[data]
webdataset<2,>=1.0

[executorch]

[executorch:python_version < "3.14"]
executorch<2.0,>=1.3

[litert]
litert-torch<1.0,>=0.9.4

[loggers]
tensorboard>=2.13.0
protobuf>=3.20.0
wandb
mlflow
clearml

[lora]
peft

[onnx]
onnx<2.0,>=1.16.0
onnxsim>=0.7.0
onnx_graphsurgeon
polygraphy

[onnx:python_version < "3.11"]
onnxruntime<1.24

[onnx:python_version >= "3.11"]
onnxruntime

[openvino]
openvino<2027.0,>=2024.0

[plus]
rfdetr_plus<2.0.0,>=1.0.1

[tensorrt]
onnxruntime-gpu
tensorrt>=8.6.1
polygraphy
onnxconverter-common
onnx<2.0,>=1.16.0

[tensorrt-bench]
pycuda>=2024.1

[tflite]

[tflite:python_version >= "3.12" and python_version < "3.13"]
onnx<2.0,>=1.20.0
onnxsim>=0.7.0
onnx_graphsurgeon
polygraphy
onnx2tf<3.0.0,>=2.4.0
flatbuffers>=23.5.26
tf-keras>=2.16.0
tensorflow>=2.16.0

[train]
peft
pytorch_lightning!=2.6.2,!=2.6.3,<3,>=2.6
torchmetrics[detection]<1.9.0,>=1.8.2
hotcoco==1.0.1
faster-coco-eval>=1.7.2
ultrafast-pycocotools<0.2,>=0.1.10
vernier<0.6,>=0.5.3
pycocotools
scipy
simplejpeg<2,>=1.9
torch-hungarian==0.1.0rc0
roboflow

[visual]
matplotlib
pandas
seaborn

[xla]

[xla:sys_platform == "linux"]
torch==2.9.*
torch_xla==2.9.*
