pypi/ai-tune-diffusion-jit-tuning-example