Allow environment variable to set number of threads to use

[jchodera]

We should add a feature to our training notebooks to check if an environment variable is set (e.g. KINOML_NUM_THREADS), and if so, call `torch.set_num_threads() with the appropriate number of threads. This will allow us to better work within batch queue systems.