Add the documentation for the torch_em.loss and top-level files

Add the documentation for the torch_em.loss and top-level files #2180