Make torch installation optional

# Description

It could be convenient to make `torch` an optional dependency to have lighter dependencies (and hence faster installations) for the users that are not interested in `torch`-related functionalities of the repository.