Add keyword use_device to CHGNet.load()

Add keyword use_device to CHGNet.load() #381