transformers>=4.25.1 flax optax torch torchvision ftfy tensorboard modelcards