Metadata-Version: 2.4
Name: nvidia-tao-automl
Version: 7.0.1
Summary: NVIDIA TAO AutoML — standalone hyperparameter optimization for TAO networks
License: Apache-2.0
Project-URL: Repository, https://gitlab-master.nvidia.com/nvidia-tao-toolkit/tao-automl
Requires-Python: >=3.10
License-File: LICENSE
Requires-Dist: numpy
Requires-Dist: scikit-learn
Requires-Dist: scipy
Requires-Dist: pandas
Requires-Dist: omegaconf
Requires-Dist: openai>=1.0
Requires-Dist: requests
Requires-Dist: PyYAML>=6.0
Requires-Dist: nvidia-tao-sdk[all]==7.0.1
Provides-Extra: lepton
Requires-Dist: nvidia-tao-sdk[all]==7.0.1; extra == "lepton"
Provides-Extra: slurm
Requires-Dist: nvidia-tao-sdk[all]==7.0.1; extra == "slurm"
Provides-Extra: kubernetes
Requires-Dist: nvidia-tao-sdk[all]==7.0.1; extra == "kubernetes"
Provides-Extra: docker
Requires-Dist: nvidia-tao-sdk[all]==7.0.1; extra == "docker"
Provides-Extra: brev
Requires-Dist: nvidia-tao-sdk[all]==7.0.1; extra == "brev"
Provides-Extra: llm
Requires-Dist: openai>=1.0; extra == "llm"
Provides-Extra: wandb
Requires-Dist: wandb>=0.15; extra == "wandb"
Provides-Extra: all
Requires-Dist: nvidia-tao-automl[brev,docker,kubernetes,lepton,llm,slurm,wandb]; extra == "all"
Provides-Extra: dev
Requires-Dist: nvidia-tao-automl[all]; extra == "dev"
Requires-Dist: pytest>=7.0; extra == "dev"
Requires-Dist: pytest-cov; extra == "dev"
Dynamic: license-file
