gradio numpy matplotlib torch torchaudio transformers stable_baselines3 gymnasium vmas datasets