python-chess scikit-learn torch pandas pandas-stubs matplotlib safetensors