transformers>=4.25.1 datasets flax optax torch torchvision ftfy tensorboard Jinja2 diffusers