Exploration of using pytorch for model fitting
conda env create -f environment.yml
If you have an existing torchexp environment that you want to update to match changes in environment.yml, you can do so with the following command:
conda env update --file environment.yml --prune