librosa ffmpeg dataclasses pathlib tqdm numpy tensorflow datasets transformers flax jax optax