Complete readme

[nmd2k]

🚀 Feature

I want to complete the readme with a guide line for setting up the package and basic usage.

Motivation

It is a lack of guidelines to use and install the package

Additional context

Follow this guideline: https://github.com/PyTorchLightning/lightning-bolts