openteamsinc
pypi/aitune
Risk Profile
Package
Source
Risk Profile
Maturity:
Mature
Health:
Healthy
Legal:
Caution Needed
License includes a patent grant clause
Security:
Healthy
Package
pypi
aitune
Version
0.6.0
Last Release Date
about 2 months ago
License
Apache
Dependencies
autowrapt
>=1.0
dill
>=0.3.8
nvidia-ml-py
nvidia-modelopt
[onnx]
>=0.33.0
numpy
>=1.26.4
onnxscript
>=0.2.0
onnx-ir
>=0.1.11
nvtx
pandas
>=2.3
psutil
tabulate
>=0.9.0
tensorrt
>=10.3
torch
>=2.8a0
torchao
>=0.12
torch-tensorrt
>=2.8a0
wrapt
~=2.0
dynamo:
ai-dynamo-runtime
>=0.1
uvloop
torch28:
torch
<2.9, >=2.8.0a0
torch-tensorrt
<2.9, >=2.8.0a0
torchao
<0.13, >=0.12
torch29:
torch
<2.10, >=2.9.0a0
torch-tensorrt
<2.10, >=2.9.0a0
torchao
<0.15, >=0.14
torch210:
torch
<2.11, >=2.10.0a0
torch-tensorrt
<2.11, >=2.10.0a0
torchao
<0.16, >=0.15
nvidia-cuda-runtime-cu13
==0.0.0a0
torch211:
torch
<2.12, >=2.11.0a0
torch-tensorrt
<2.12, >=2.11.0a0
torchao
<0.17, >=0.16
torch212:
torch
<2.13, >=2.12.0a0
torch-tensorrt
<2.13, >=2.12.0a0
torchao
<0.18, >=0.17
Source
Location
ai-dynamo/aitune
Last Source Update
3 days ago
Licenses
Apache License 2.0
Distribution Destinations
pypi/aitune
pypi/ai-tune-llm-example
pypi/ai-tune-wan-example
pypi/aitune-bert-triton-example
pypi/ai-tune-esm2-example
pypi/ai-tune-flux-example
pypi/aitune-yolo-triton-example
pypi/aitune-resnet-example
pypi/aitune-examples-common
pypi/nvidia-ai-tune-e5-large-v2-embedding-example
pypi/ai-tune-diffusion-jit-tuning-example
pypi/aitune-parakeet-ctc-example
pypi/aitune-parakeet-rnnt-example
pypi/ai-tune-stable-diffusion-example
