Metadata-Version: 2.1
Name: nvidia-ammo
Version: 0.9.1
Summary: Ammo: a unified algorithmic model optimization and deployment toolkit.
Author-email: "Nvidia, Inc." <ammo-support@exchange.nvidia.com>
License: NVIDIA Proprietary Software
Project-URL: Homepage, https://nvidia.com
Classifier: Programming Language :: Python :: 3
Classifier: Operating System :: OS Independent
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Intended Audience :: Science/Research
Requires-Python: <3.12,>=3.8
Description-Content-Type: text/x-rst
License-File: LICENSE
Requires-Dist: ninja
Requires-Dist: networkx
Requires-Dist: numpy
Requires-Dist: onnx >=1.14.0
Requires-Dist: onnxruntime ~=1.16.1
Requires-Dist: onnx-graphsurgeon
Requires-Dist: packaging
Requires-Dist: pydantic >=2.0
Requires-Dist: rich
Requires-Dist: scipy
Requires-Dist: torch >=1.11
Requires-Dist: tqdm
Requires-Dist: transformers
Provides-Extra: deploy
Requires-Dist: mpi4py ; extra == 'deploy'
Provides-Extra: dev
Requires-Dist: black ==23.3.0 ; extra == 'dev'
Requires-Dist: mypy ==1.3.0 ; extra == 'dev'
Requires-Dist: pre-commit ==3.3.2 ; extra == 'dev'
Requires-Dist: ruff ==0.0.272 ; extra == 'dev'
Requires-Dist: accelerate ; extra == 'dev'
Requires-Dist: coverage ; extra == 'dev'
Requires-Dist: onnxsim ; extra == 'dev'
Requires-Dist: parameterized ; extra == 'dev'
Requires-Dist: pytest ; extra == 'dev'
Requires-Dist: pytest-asyncio ; extra == 'dev'
Requires-Dist: pytest-cov ; extra == 'dev'
Requires-Dist: pytest-timeout ; extra == 'dev'
Requires-Dist: tox ==4.5.2 ; extra == 'dev'
Requires-Dist: tox-current-env ; extra == 'dev'
Requires-Dist: furo >=2023.5.20 ; extra == 'dev'
Requires-Dist: nbsphinx >=0.9.2 ; extra == 'dev'
Requires-Dist: pypandoc ; extra == 'dev'
Requires-Dist: sphinx >=7.0.1 ; extra == 'dev'
Requires-Dist: sphinx-autobuild >=2021.3.14 ; extra == 'dev'
Requires-Dist: sphinx-copybutton >=0.5.2 ; extra == 'dev'
Requires-Dist: sphinx-inline-tabs >=2023.4.21 ; extra == 'dev'
Requires-Dist: sphinx-togglebutton >=0.3.2 ; extra == 'dev'
Requires-Dist: autodoc-pydantic ; extra == 'dev'
Requires-Dist: cython ; extra == 'dev'
Requires-Dist: setuptools >=67.8.0 ; extra == 'dev'
Requires-Dist: setuptools-scm >=7.1.0 ; extra == 'dev'
Requires-Dist: toml ; extra == 'dev'
Requires-Dist: twine ; extra == 'dev'
Requires-Dist: ipython ; extra == 'dev'
Requires-Dist: ipywidgets ; extra == 'dev'
Requires-Dist: timm ; extra == 'dev'
Requires-Dist: torchvision ; extra == 'dev'
Requires-Dist: opencv-python ; extra == 'dev'
Requires-Dist: pandas ; extra == 'dev'
Requires-Dist: matplotlib ; extra == 'dev'
Requires-Dist: seaborn ; extra == 'dev'
Requires-Dist: bandit[toml] ==1.7.5 ; extra == 'dev'
Provides-Extra: dlfw
Requires-Dist: ninja ; extra == 'dlfw'
Requires-Dist: networkx ; extra == 'dlfw'
Requires-Dist: numpy ; extra == 'dlfw'
Requires-Dist: packaging ; extra == 'dlfw'
Requires-Dist: pydantic >=2.0 ; extra == 'dlfw'
Requires-Dist: rich ; extra == 'dlfw'
Requires-Dist: scipy ; extra == 'dlfw'
Requires-Dist: torch >=1.11 ; extra == 'dlfw'
Requires-Dist: tqdm ; extra == 'dlfw'
Provides-Extra: onnx
Requires-Dist: jax ; extra == 'onnx'

AMMO
#########

*AlgorithMic Model Optimization.*

--------

AMMO is a library for conveniently optimizing and deploying efficient neural networks that can
fit a wide range of Nvidia hardware. It is intended for ML engineers to efficiently design, train,
and deploy their models on Nvidia from within their desired ML training framework, e.g., PyTorch.
